AtCoder Beginner Contest 207E题Mod i（前缀和优化dp）

题目链接
大概意思是把A序列分成连续的K段，第i段的和要整除i
问总的方案数。
emmm 很容易想到定义dp[i][j]为前i个数分成j段的方案数
怎么转移呢，考虑同余，显然dp[i][j]=Σ [(sum_i-sum_p)%j==0]dp[p][j-1]
然后就成了N³ 炸裂。
不难发现，sum_i 与sum_p 同余， 因此我们可以利用同余性质做个前缀和
即pre[i][j] 表示分为j组时，同余为i时的方案，那么dp[i][j] =pre[sum_i%j][j]
N²

 #include<bits/stdc++.h>
#include<stdlib.h>
#include<algorithm>
#include<stdio.h>
#include<string.h>
#include<queue>
#include<time.h>
#include <cstdio>
#include <iostream>
#include <vector>
#define ll long long
#define int long long
#define inf 0x3f3f3f3f
#define mods 1000000007
#define modd 998244353
#define PI acos(-1)
#define fi first
#define se second
#define lowbit(x) (x&(-x))
#define mp make_pair
#define pb push_back
#define si size()
#define E exp(1.0)
#define fixed cout.setf(ios::fixed)
#define fixeds(x) setprecision(x)
#define IOS ios::sync_with_stdio(false);cin.tie(0)
 using namespace std;
 ll gcd(ll a,ll b){if(a<0)a=-a;if(b<0)b=-b;return b==0?a:gcd(b,a%b);}
template<typename T>void read(T &res){bool flag=false;char ch;while(!isdigit(ch=getchar()))(ch=='-')&&(flag=true);
for(res=ch-48;isdigit(ch=getchar());res=(res<<1)+(res<<3)+ch - 48);flag&&(res=-res);}
ll lcm(ll a,ll b){return a*b/gcd(a,b);}
ll qp(ll a,ll b,ll mod){ll ans=1;if(b==0){return ans%mod;}while(b){if(b%2==1){b--;ans=ans*a%mod;}a=a*a%mod;b=b/2;}return ans%mod;}//快速幂%
ll qpn(ll a,ll b, ll p){ll ans = 1;a%=p;while(b){if(b&1){ans = (ans*a)%p;--b;}a =(a*a)%p;b >>= 1;}return ans%p;}//逆元   (分子*qp(分母,mod-2,mod))%mod;

ll a[4444];
ll dp[3333][3333];
ll s[3333][3333];
ll add[4444];
signed main()
{

    ll n;
    read(n);
    for(int i=1; i<=n; i++)
    {
        read(a[i]);
        //add[i]+=a[i-1];
        add[i]=add[i-1]+a[i];
    }
    s[0][1]=1;
    for(int i=1; i<=n; i++)
    {

        for(int j=1; j<=i; j++)
        {
            dp[i][j]=s[add[i]%j][j]%mods;

        }

        for(int j=1; j<=i+1; j++)
        {
            s[add[i]%j][j]=(s[add[i]%j][j]+dp[i][j-1] )%mods;

        }

    }
    ll ans=0;
   for(int i=1;i<=n;i++){
   ans=(ans+dp[n][i])%mods;

   }
printf("%lld",ans);

}