赛时代码,本来思路就没有很清晰,过了几天就忘干净了。
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=200004,mod=998244353;
int T,n,a[maxn],f[2][maxn],g[2][maxn];
int _abs(int x){
return x<0?-x:x;
}
void solve(){
scanf("%lld",&n);
f[0][0]=f[1][0]=0;
g[0][0]=g[1][0]=1;
for(int i=1;i<=n;i++){
scanf("%lld",&a[i]);
f[0][i]=max(f[0][i-1]+a[i],_abs(f[1][i-1]+a[i]));
if(f[0][i]==f[0][i-1]+a[i]){
g[0][i]=(g[0][i]+g[0][i-1])%mod;
if(f[0][i-1]+a[i]>=0&&f[0][i-1]!=f[1][i-1])
g[0][i]=(g[0][i]+g[0][i-1])%mod;
}
if(f[0][i]==_abs(f[1][i-1]+a[i]))
g[0][i]=(g[0][i]+g[1][i-1])%mod;
f[1][i]=min(f[1][i-1]+a[i],_abs(f[1][i-1]+a[i]));
if(f[1][i]==f[1][i-1]+a[i])
g[1][i]=(g[1][i]+g[1][i-1])%mod;
if(f[1][i]==_abs(f[1][i-1]+a[i]))
g[1][i]=(g[1][i]+g[1][i-1])%mod;
}
printf("%lld\n",g[0][n]);
for(int i=0;i<=n;i++){
f[0][i]=g[0][i]=0;
f[1][i]=g[1][i]=0;
}
}
signed main(){
scanf("%d",&T);
while(T--)
solve();
return 0;
}
/*
1
6
-1 1 -4 5 -1 4
*/
想问问这种要分多种情况讨论的 dp 怎么样做可以让自己的思路更清晰?