ABC207 E. Mod i DP前缀和优化

题目链接

题解：

让$sum[i]$表示前$i$个数的和，$dp[i][j]$表示将前$i$个数分成$j$段的可行方案数。

接下来，找状态转移方程：

$dp[i][j]+=dp[k][j-1]$，其中满足$(sum[i]-sum[k])\%j==0$。

这样直接枚举，可以发现时间复杂度是$O(n^3)$，会超时，因此需要优化。

我们可以发现，对于$(sum[i]-sum[k])\%j==0$，这个条件，其实可以转化为$sum[i]\%j==sum[k]\%j$。

那么，对于转移方程中的累加，我们可以使用前缀和优化解决。也就是在转移的过程中，处理出满足$sum[i]\%j==sum[k]\%j$的所有位置的$dp[k][j-1]$的和。

这样程序便优化到$O(n^2)$。

实现细节见代码：

#include <bits/stdc++.h>
using namespace std;
#define endl '\n'
#define int long long
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int MAXN = 3e3 + 10;
const int mod = 1e9 + 7;
int sum[MAXN], presum[MAXN][MAXN], dp[MAXN][MAXN];
signed main() 
{
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
    int n;
    cin >> n;
    for (int i = 1; i <= n; i++) {
        int a;
        cin >> a;
        sum[i] = sum[i - 1] + a;
    }
    for (int i = 1; i <= n; i++) {
        dp[i][1] = 1;
        for (int j = 2; j <= i; j++) {
            int now = (sum[i] % j);
            dp[i][j] = presum[j][now];
        }
        for (int j = 1; j <= n; j++) {
            int now = sum[i] % j;
            presum[j][now] = (presum[j][now] + dp[i][j - 1]) % mod; // 处理前缀和
        }
    }
    int ans = 0;
    for (int i = 1; i <= n; i++) {
        ans = (ans + dp[n][i]) % mod;
    }
    cout << ans << endl;
    return 0;
}