[ABC159F] Knapsack for All Segments题解

题目传送门

题目传送门（洛谷）
题目传送门（atcoder）

Step1 理解题意

• 给定一个 $n$ 和一个$s$
• 给定一个长度为 $n$ 的正整数数组 $a$
• 定义函数 $f(L,R)$ 是指在 $L$ 到 $R$ 的区间内能使其中子串（子串是不需要连续的）之和为 $s$ 的子串数量
• 求在 $a$ 中所有 $f(L,R)$ 之和
• 结果对 $998244353$ 取模
• $1\le N,S,A_i\le 3000$
  

Step2 思路解释

让 $d{p}_{i,j}$指在前 $i$ 个数的总和为 $j$,那么我们易得以下转移方程：
$d{p}_{i,j}\leftarrow (d{p}_{i-1,j}+d{p}_{i-1,j-a_i})\mathrm{mod}p$

Step3 简化数组

• 因为每次枚举 $d{p}_{i,j}$都只要用到 $d{p}_{i-1,x}$，所以我们可以省去一维，即用 $d{p}_i$ 表示 $d{p}_{i,j}$。
• 如果只考虑 $A_i=S$ 的情况，则会少算 $i$ 种右端点为 $i$ 的方案，所以对于每个 $i$，应该使$d{p}_0\leftarrow i$。

Step4 Ac code

#include<bits/stdc++.h>
#define int long long
using namespace std;

const int mod = 998244353;

int a[3005],dp[3005];
int ans;
signed main(){
	int n ,s;
	scanf("%lld%lld",&n,&s);
	for(int i = 1; i <= n; i++) scanf("%lld",&a[i]);
	for(int i = 1; i <= n ;i++){
		dp[0]++;
		for(int j = s; j >= a[i]; j--){
			dp[j] = (dp[j] + dp[j - a[i]]) % mod; 
		}
		ans += dp[s];
		ans %= mod;
	}
	printf("%lld",ans);
	return 0;
}