D - I Hate Non-integer Number （选数的计数dp

__Rain

于 2022-08-01 07:47:18 发布

阅读量444

点赞数

CC 4.0 BY-SA版权

分类专栏：线性dp 文章标签：算法图论 c++

本文链接：https://blog.csdn.net/cosx_/article/details/126093515

线性dp 专栏收录该内容

21 篇文章

订阅专栏

该博客介绍了一种使用动态规划求解数组中选择特定数量的元素，使得它们的和对某数取模后的结果为零的算法。代码中展示了如何初始化状态转移矩阵，并通过迭代更新来获取所有可能的选择方案数。最终，博客给出了计算所有模数情况下的总方案数并输出结果的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

D - I Hate Non-integer Number
思路：
枚举选 $l$ 个数，然后 $d p [i] [j] [k]$ 表示前 $i$ 个数选 $j$ 个数 $\%l$ 的和为 $k$ 的方案数
那么答案就是所有 $l$ 情况下的 $d p [n] [l] [0]$ 的加和
code：

#include<bits/stdc++.h>
#define endl '\n'
#define ll long long
#define ull unsigned long long
#define ld long double
#define all(x) x.begin(), x.end()
#define mem(x, d) memset(x, d, sizeof(x))
#define eps 1e-6
using namespace std;
const int maxn = 2e6 + 9;
const int mod = 998244353;
const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
ll n, m;
int a[109];
ll dp[109][109][109];
//  前i个数选j个，和 %l 为 k 的方案数 
ll ans = 0;
void work()
{
	cin >> n;
	for(int i = 1; i <= n; ++i) {
		cin >> a[i];
	}
	for(int l = 1; l <= n; ++l)
	{
		mem(dp, 0);
		dp[0][0][0] = 1;
		for(int i = 1; i <= n; ++i){
			for(int j = 0; j <= i; ++j)
				for(int k = 0; k < l; ++k)
					dp[i][j][k] = dp[i-1][j][k];// 先把不选a_i的情况转移一下
			for(int j = 0; j <= i; ++j){
				for(int k = 0; k < l; ++k){
					if(j >= 1){
						int sum = (a[i] + k) % l;
						(dp[i][j][sum] += dp[i-1][j-1][k]) %= mod;
					}
				}
			}
		}
		(ans += dp[n][l][0]) %= mod;
	}
	cout << ans;
}

int main()
{
	ios::sync_with_stdio(0);
//	int TT;cin>>TT;while(TT--)
	work();
	return 0;
}