AtCoder Beginner Contest 159 F.Knapsack for All Segments

题目链接
思路：考虑每个和为s的组合 的贡献。
显然贡献为L*(n-R+1) ,L,R 为组合的左右两端的位置。
如果枚举当前R位置 ，那么只要我们知道所有合法情况的L值之和那么就可以统计出答案了。
考虑dp。
$dp[i][j]$为第i位，和为j的L之和
每个数 $a_i$当成起点时初始 $dp[i]~ [{a_i}]=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;
}