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

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

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

所以 $a 1, a 2$ 按照 $a_i$ 从大到小排序

枚举 $a 1$ 数组取前 $i$ 个物品

此时可以使用二分去快速得到一个最小的 $i n d e x$ 使得取走前 $i n d e x$ 大的 $a 2$ 元素

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

~~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;
	}
}