ABC207 E - Mod i（dp优化）

题意：

解法：

令sum[i]表示前i个数的和.
令d[i][j]表示前i个数分成j段的方案数.

转移方程:
d[i][j]+=d[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[k]%j=sum[i]%j的所有位置的d[k][j-1]的和即可,
令ms[j][t]表示sum[i]%j=t的d[i][j-1]的和,dp过程中维护一下即可.

然后就优化成O(n^2)的了.

code：

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxm=3e3+5;
const int mod=1e9+7;
int ms[maxm][maxm];
int d[maxm][maxm];//d[i][j]表示前i个数分成j段的方案数
int sum[maxm];
int a[maxm];
int n;
void solve(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        sum[i]=sum[i-1]+a[i];
    }
    for(int i=1;i<=n;i++){
        d[i][1]=1;
        for(int j=2;j<=i;j++){
            int t=sum[i]%j;
            d[i][j]=ms[j][t];
        }
        for(int j=1;j<=n;j++){
            int t=sum[i]%j;
            ms[j][t]=(ms[j][t]+d[i][j-1])%mod;
        }
    }
    int ans=0;
    for(int i=1;i<=n;i++){
        ans=(ans+d[n][i])%mod;
    }
    cout<<ans<<endl;
}
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);
    solve();
    return 0;
}

---