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

傳送門

因為 b i b_i bi?只有 1 1 1和 2 2 2兩種,考慮分成 a 1 , a 2 a1,a2 a1,a2兩個陣列裝相應的 a i a_i ai?

a 1 a1 a1里的 a i a_i ai?的 b i = 1 b_i=1 bi?=1,那么 a 1 a1 a1里的物品具有明顯的優劣關系, a i a_i ai?越大的越好

所以 a 1 , a 2 a1,a2 a1,a2按照 a i a_i ai?從大到小排序

列舉 a 1 a1 a1陣列取前 i i i個物品

此時可以使用二分去快速得到一個最小的 i n d e x index index使得取走前 i n d e x index index大的 a 2 a2 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;
	}
}