I Hate Non-integer Number（DP）

题意

有一个包含\(N\)个元素的数组\(A\).

有\(2^N - 1\)种方式从中选择至少一项。问其中有多少满足平均值为整数。

题目链接：https://atcoder.jp/contests/abc262/tasks/abc262_d

数据范围

\(1 \leq N \leq 100\)

思路

如果选中了\(x_1,x_2,\dots, x_i\)，那么它们的平均值为\(\frac{x_1 + \dots + x_i}{i}\)。这个数值为整数，当且仅当选中项之和为\(i\)的倍数。

在这里，我们考虑对\(i=1,2,\dots, N\)分别进行DP。

令\(f(j,k,l)\)表示从前\(j\)个元素中选\(k\)项，并且模\(i\)余\(l\)的方案数。转移过程显然。

最终答案为\(\sum\limits_{i=1}^N f(N,i,0)\)

代码

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>

using namespace std;

typedef long long ll;

const int N = 110, mod = 998244353;

int n;
ll a[N];
ll f[N][N][N];

int main()
{
    scanf("%d", &n);
    for(int i = 1; i <= n; i ++) scanf("%lld", &a[i]);
    ll res = 0;
    for(int i = 1; i <= n; i ++) {
        memset(f, 0, sizeof f);
        f[0][0][0] = 1;
        for(int j = 0; j < n; j ++) {
            for(int k = 0; k <= i; k ++) {
                for(int l = 0; l < i; l ++) {
                    f[j + 1][k][l] = (f[j + 1][k][l] + f[j][k][l]) % mod;
                    ll t = (l + a[j + 1]) % i;
                    if(k != i) f[j + 1][k + 1][t] = (f[j + 1][k + 1][t] + f[j][k][l]) % mod;
                }
            }
        }
        res = (res + f[n][i][0]) % mod;
    }
    printf("%lld\n", res);
    return 0;
}