1475D. Cleaning the Phone(二分+前缀和)

Posted 2023-02-24 issue是fw

传送门

因为$b_i$只有$1$和$2$两种,考虑分成$a1,a2$两个数组装相应的$a_i$

$a1$里的$a_i$的$b_i=1$,那么$a1$里的物品具有明显的优劣关系,$a_i$越大的越好

所以$a1,a2$按照$a_i$从大到小排序

枚举$a1$数组取前$i$个物品

此时可以使用二分去快速得到一个最小的$index$使得取走前$index$大的$a2$元素

满足条件,然后更新答案即可

dp似乎是行不通的(可能可以???)…

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int maxn = 3e5+10;
const int inf = 1e15;
int t,n,m,a[maxn],b[maxn],a1[maxn],a2[maxn],pre[maxn];
bool com(int a,int b) return a>b; 
signed main()

	int t; cin >> t;
	while( t-- )
	
		cin >> n >> m;
		for(int i=1;i<=n;i++)	scanf("%lld",&a[i]);
		for(int i=1;i<=n;i++)	scanf("%lld",&b[i]); 
		a1[0]=a2[0]=0;
		for(int i=1;i<=n;i++)
		
			if( b[i]==1 )	a1[++a1[0]] = a[i];
			else	a2[++a2[0]] = a[i];
		
		sort( a1+1,a1+1+a1[0],com );
		sort( a2+1,a2+1+a2[0],com );
		for(int i=1;i<=a2[0];i++)	pre[i] = pre[i-1]+a2[i];
		//ö��ѡ��s��
		int ans = inf,sum=0;
		for(int i=1;i<=a1[0];i++)
		
			sum+=a1[i];
			if( sum>=m ) ans=min(i,ans); break; 
			int yu = m-sum;
			int index = lower_bound( pre+1,pre+1+a2[0],yu )-pre;
			if( index==a2[0]+1 )	continue;
			ans = min( ans,i+index*2 );
		 
		int index = lower_bound( pre+1,pre+1+a2[0],m )-pre;
		if( index!=a2[0]+1 )
			ans = min( ans,index*2 );
		if( ans==inf )	cout << -1 << endl;
		else	cout << ans << endl;