AtCoder Beginner Contest 207 E - Mod i(dp)

题意：

给出一个长度为$n$的数组$a$，假如将这个数组分成$k$个连续段，第$i$段的和为$b_i$,如果对于每一个$i$，$b_i\%i==0$,那么该分割方法称为合法的，求有多少个合法的分割方法。

题解：

假如不考虑复杂度，那么可以想出一个$O(n^3)$的做法：设$dp[i][j]$表示前$i$个数分为$j$段的合法方案数，那么转移方程就是

$dp[i][j]+=dp[k][j-1]$ $(sum(k+1,i)\%j==0)$

考虑怎么优化，因为$sum(k+1,i)\%j==0$ 所以 $(sum(1,i)-sum(1,k))\%j==0$

又可以推出$sum(1,i)\%j==sum(1,k)\%j$

那么我们可以先枚举$j$，再枚举$i$，每次将$dp[i][j-1]$放入数组$num[sum(1,i)\%j]$, 那么$dp[i][j]=num[sum(1,i)\%j]$

为什么呢？上面$n^3$的转移式中，将合适位置的$k$的$dp[k][j-1]$算入贡献，而$num$数组里面存的正是所有满足要求的$dp[k][j-1]$的和。

代码：

#pragma GCC diagnostic error "-std=c++11"
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<stack>
#include<set>
#include<ctime>
#define iss ios::sync_with_stdio(false)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef pair<int,int> pii;
const int mod=1e9+7;
const int MAXN=3e3+5;
const int inf=0x3f3f3f3f;
int n;
ll a[MAXN];
ll sum[MAXN];
ll dp[MAXN][MAXN];
ll f[MAXN][MAXN];
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        sum[i]=sum[i-1]+a[i];
    }
    f[0][0]=1;
    for(int j=1;j<=n;j++)
    {
        for(int i=1;i<=n;i++)
        {
            dp[i][j]=f[sum[i]%j][j-1];
            f[sum[i]%j][j-1]=(f[sum[i]%j][j-1]+dp[i][j-1])%mod;
        }
    }
    ll ans=0;
    for(int i=1;i<=n;i++)
    {
        ans=(ans+dp[n][i])%mod;
    }
    cout<<ans<<endl;
}