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

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

#include
#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;
}

你可能感兴趣的:(线性dp,算法,图论,c++)