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

谁借莪１个温暖的怀抱￠ 2024-03-17 16:03 79阅读 0赞

[F - Max Sum Counting (atcoder.jp)][F - Max Sum Counting _atcoder.jp]

题意：

![0ad34ab47cc542cfb0e62edec82976b2.png][]

思路：

首先求方案数，除了组合数就是计数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;
    }

[F - Max Sum Counting _atcoder.jp]: https://atcoder.jp/contests/abc216/tasks/abc216_f
[0ad34ab47cc542cfb0e62edec82976b2.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/15/c393dc9f115b41039c8792217b7ab276.png