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;
}