CodeForces 1475D - Cleaning the Phone(枚举+二分)

传送门：https://codeforces.com/problemset/problem/1475/D

题面：

题意:

手机中有n个app，现在需要腾出m的内存空间
每个app占用内存$a_i$，重要程度$b_i$（也就可以理解为代价吧）
要求腾出m的内存空间所需要的最小代价

思路：

弄懂题意后好像就是一道01背包问题，
写到一半发现$m\in [1,10^9]$，只能另辟蹊径了。。

---

因为$b_i$就只有两种情况{1，2}，
我们把app的内存分到代价为1或代价为2两种情况的数组v1,v2里去
考虑到最优解，一定是相同代价的情况下先删较大内存的app，所以对两个分组从大到小排序

然后枚举v1中前i个app的内存和$sum1$，（枚举v2也可以，一样的道理）
那么还需要释放$m-sum1$的内存从v2中取，
二分v2的前缀和sum2，找到释放$m-sum1$的内存的最少app数目index2
最后的答案就是 m i n { i + i n d e x 2 ∗ 2 } min\{i+index2*2\} min{i+index2∗2}

注意v1的大小为0的情况，也就是只有代价为2的app，那么不会进入v1的枚举循环
要特判这种情况

AC代码:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+10,inf=0x3f3f3f3f;
int n,m,t,v1[maxn],v2[maxn],a[maxn],b;
ll sum2[maxn];
int main() {
	cin>>t;
	for(int k=1; k<=t; k++) {
		ll suma=0,len1=0,len2=0;
		memset(sum2,0,sizeof(sum2));
		cin>>n>>m;
		for(int i=1; i<=n; i++) cin>>a[i],suma+=a[i];
		for(int i=1; i<=n; i++) {
			cin>>b;
			if(b==1) v1[++len1]=a[i];
			else v2[++len2]=a[i];
		}
		if(suma<m) {
			printf("-1\n");
			continue;
		}
		sort(v1+1,v1+1+len1,greater<int>());
		sort(v2+1,v2+1+len2,greater<int>());
 
		//for(int i=1; i<=len1; i++) cout<<v1[i]<<(i==len1 ? "\n" : " ");
		//for(int i=1; i<=len2; i++) cout<<v2[i]<<(i==len2 ? "\n" : " ");
 
		ll ans=inf,sum1=0;
		
		for(int i=1; i<=len2; i++) sum2[i]=sum2[i-1]+v2[i];
		for(int i=1; i<=len1; i++) {
			sum1=sum1+v1[i];
			if(sum1>=m) {
				ans=min((ll)i,ans);//代价b
				//printf("===sum1=%d",sum1);
				break;
			}
			ll index2=lower_bound(sum2+1,sum2+1+len2,m-sum1)-sum2;
			if(index2==len2+1) continue;
			ans=min(ans,i+index2*2);
			//printf("index1=%d index2=%d ans=%d\n",i+1,index2,ans);
		}
		// 如果 len1==0 的情况
		ll index2=lower_bound(sum2+1,sum2+1+len2,m)-sum2;
		if(index2 != len2+1) ans = min(ans,index2*2);
		printf("%lld\n",ans==inf? -1 : ans);
		//printf("========%d : %lld\n",k,ans==inf? -1 : ans);
 
	}
}