cf 1485 F Copy or Prefix Sum 奇怪的dp

link

题意

• 有一数组$B$，求满足以下条件的数组$A$的个数$$\begin{gathered} B[i]=A[i]，OR \\ B[i]=\sum _{j=1}^{i}A[j] \end{gathered}$$

题解

• 第一眼看到以为是啥自己没学过的知识点，但其实往dp方面想，很容易想到是通过存储前缀和来转移，又由于是算方案数。所以有
• $dp[i][j]$表示${\sum }_{k=1}^{i}A[k]=j$的方案数，那么显然有$n^2{\log }_2^n$的转移方程$$\begin{gathered} dp[i][j]=dp[i-1][j-b[i]] \\ dp[i][b[i]]=\sum _{j=-inf}^{inf}[j!=0]dp[i-1][j] \end{gathered}$$
• 对于方程二来说其实只需要维护一下前缀总方案数就可以$O(1)$转移
• 对于方程一也能看出其实是一个整体右移的转移，那么我们只需要维护一下偏移量就可以$O(1)$转移啦

#include<bits/stdc++.h>
#define mp make_pair
#define pb push_back
#define SZ(x) (int)(x.size())
#define all(x) x.begin(),x.end()
using namespace std;
typedef long long ll;
typedef pair<int,int>pii;
const int maxn = 1e6 + 6;
const int inf = 0x3f3f3f3f;
const int mod = 1e9 + 7;
const int N = 2e3 + 3;
ll qpow(ll x,ll y) { ll ans=1;x%=mod;  while(y) { if(y&1) ans=ans*x%mod; x=x*x%mod;  y>>=1; } return ans; }

int _,n;
ll b[maxn];
map<ll,ll>M;
int main() {
    for(scanf("%d",&_);_;_--) { 
        M.clear();
        scanf("%d",&n);
        for(int i=1;i<=n;i++) scanf("%lld",&b[i]);
        ll sum=1,hor=0;
        M[0]=1;
        for(int i=1;i<=n;i++) {
            ll ch=(sum+mod-M[-hor])%mod;
            hor+=b[i];
            M[b[i]-hor]=(M[b[i]-hor]+ch)%mod;
            sum=(sum+ch)%mod;
        }       
        printf("%lld\n",sum);
    }
    return 0;
}