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