【排序+背包求方案数】ABC216 F

F - Max Sum Counting (atcoder.jp)

题意：

思路：

首先求方案数，除了组合数就是计数DP

但是这道题数据范围是5000，因此考虑DP

因为求的是子集的方案数，因此可以考虑给数组排序

排完序之后做一次背包（子集用背包求）

它的转移条件是A数组中选的最大值>=B数组中选的数的和

这个条件在转移的时候体现

DP中的限制条件除了多一维，也可以在转移的时候加条件

因为已经排好序了，因此选的Ai一定是最大值

那么只需要对B数组做一次01背包即可

Code：

#include <bits/stdc++.h>

#define int long long

using namespace std;

const int mxn=5e3+10;
const int mxe=1e6+10;
const int mod=998244353;
const int Inf=0x3f3f3f3f;

struct ty{
	int a,b;
}p[mxn];

int N;
int dp[mxn];
//选了b数组子集为j的方案数

bool cmp(ty x,ty y){
	return x.a<y.a;
}
void solve(){
	cin>>N;
	for(int i=1;i<=N;i++) cin>>p[i].a;
	for(int i=1;i<=N;i++) cin>>p[i].b;
	sort(p+1,p+1+N,cmp);
	dp[0]=1;
	int ans=0;
	for(int i=1;i<=N;i++){
		for(int j=p[i].b;j<=p[i].a;j++){
			ans+=dp[j-p[i].b];
			ans%=mod;
		}
		for(int j=5000;j>=p[i].b;j--){
			dp[j]+=dp[j-p[i].b];
			dp[j]%=mod;
		}
	}
	cout<<ans<<'\n';
}
signed main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int __=1;//cin>>__;
	while(__--)solve();return 0;
}