AtCoder Beginner Contest 159 F Knapsack for All Segments 01背包+一维数组+排列组合

AtCoder Beginner Contest 159   比赛人数8276  快,比赛开始后3分钟看到所有题

AtCoder Beginner Contest 159 F Knapsack for All Segments  01背包+一维数组+排列组合

总目录详见https://blog.csdn.net/mrcrack/article/details/104454762

在线测评地址https://atcoder.jp/contests/abc159/tasks/abc159_f

01背包+一维数组+排列组合

处理到i个元素时，f[j]代表，对应元素和为j，必须加入a[i]元素，[1,i]元素区间内，对应的种类数量。详见代码。

AC代码如下

#include <stdio.h>
#define maxn 3010
#define mod 998244353
#define LL long long
LL a[maxn],f[maxn],ans;
int main(){
	int i,j,n,m;
	scanf("%d%d",&n,&m);
	for(i=1;i<=n;i++)scanf("%lld",&a[i]);
	for(i=1;i<=n;i++){
		for(j=m;j>=a[i];j--)f[j]=(f[j]+f[j-a[i]])%mod;
		f[a[i]]=(f[a[i]]+i)%mod;
		ans=(ans+f[m]*(n-i+1))%mod;//f[m]代表，对应元素和为m,必须加入a[i]元素，自[1,i]区间对应的种类数量
		f[m]=0;
	}
	printf("%lld\n",ans);
	return 0;
}