codeforces #697 div3 D

题目如下:
在这里插入图片描述
此题思路:
b数组实际上是一个用来分割的数组,即把a数组分成便利值为1和2的两个数组,再分别求前缀和,最后通过枚举前缀和来表示选择多少割1和多少个2。
但要注意
1,用来储存前缀和的数组要开long long。
2,使用lower_bound来加速,如果两个数组都枚举会TLE。

AC代码:

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define INF 0x3f3f3f
const int MAXN=2e5+5;
ll a[MAXN],s1[MAXN],s2[MAXN],d1[MAXN],d2[MAXN];
bool cmp(int a,int b){return a>b;}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	int T;
	cin>>T;
	while(T--)
	{
		int n,m;
		ll sum=0;
		cin>>n>>m;
		for(int i=1;i<=n;i++)
		{
			cin>>a[i];
			sum+=a[i];
		}
		int j=0,k=0;
		for(int i=1;i<=n;i++)
		{
			int op;
			cin>>op;
			if(op==1)s1[++j]=a[i];
			else s2[++k]=a[i];
		}
		if(sum<m)
		{
			cout<<-1<<endl;
			continue;
		}
		sort(s1+1,s1+1+j,cmp);
		sort(s2+1,s2+1+k,cmp);
		for(int i=1;i<=j;i++)
		{
			d1[i]=d1[i-1]+s1[i];
		}
		for(int i=1;i<=k;i++)
		{
			d2[i]=d2[i-1]+s2[i];
		}
		sum=INF;
		for(int i=0;i<=j;i++)
		{
			if(d1[i]>=m)sum=min(sum,(ll)i);
			else 
			{
			int g=lower_bound(d2+1,d2+k+1,m-d1[i])-d2;
			if(g==k+1)continue;
			sum=min(sum,(ll)(i+g*2));
		    }
		}
		cout<<sum<<endl;
	}
	return 0;
 } 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值