题目链接
思路:考虑每个和为s的组合 的贡献。
显然贡献为L*(n-R+1) ,L,R 为组合的左右两端的位置。
如果枚举当前R位置 ,那么只要我们知道所有合法情况的L值之和那么就可以统计出答案了。
考虑dp。
dp[i][j]为第i位,和为j的L之和
每个数 ai当成起点时初始 dp[i] [ai]=i
否则为0
然后剩下 就是 直接转移了。
统计答案就是 以当前位置为最右端的情况下计算答案。
小细节讨论讨论即可。
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 2e5 + 10;
#define fi first
#define se second
#define pb push_back
int n,s,a[N];
LL dp[3333][3333];
const LL mod=998244353;
int main() {
ios::sync_with_stdio(false);
cin>>n>>s;
for(int i=1;i<=n;i++){
cin>>a[i];
}
if(a[1]!=s)dp[1][a[1]]=1;LL ans=0;
if(a[1]==s)ans=n;
for(int i=2;i<=n;i++){
if(a[i]<s)dp[i][a[i]]+=i;
dp[i][a[i]]%=mod;
if(s>a[i]){
ans+=dp[i-1][s-a[i]]*(n-i+1)%mod;
}
else if(s==a[i])ans+=1ll*(n-i+1)*i%mod;
for(int j=a[i];j<s;j++){//从前面的位置转移
dp[i][j]+=dp[i-1][j-a[i]];
dp[i][j]%=mod;
}
for(int j=1;j<s;j++){
dp[i][j]+=dp[i-1][j];
dp[i][j]%=mod;
}
ans%=mod;
}
cout<<ans<<'\n';
return 0;
}