洛谷多校补题 5 B

Bolshevik

题目链接

题目大意

有n（1e6）个人你是第m个人
第i个人有ai块钱，其中bi块钱在钱包里，如果偷了第i个人的钱，他的钱就会变为ai - bi
你是一个小偷，至少投多少个人的钱才能变成这n个人里最有钱的人（可以有人跟你一样有钱）

怎么做？

贪心。。
先按ai排序，为什么？ 因为如果偷第二大ai的钱的话，最多的钱就会被第一大的限制住，所以偷钱偷的是一个前缀，然后按ai排序，O(n)枚举前缀，比如现在枚举到了第i个i+1~n得挑bi大的偷。现在只用找出来选取的前缀里最大的ai - bi和没选取的里面的最大的ai，最后要偷完的总钱得大于等于这两个的最大值。
前缀和很好弄，要找的最大值也很好弄，因为是按ai排序的，所以最大值是a[i + 1]。但是后面选取最大的b没法弄，于是就得有一个数据结构来维护，维护什么值? 看一下从第i - 1个变成第i个的变化：删除bi，查询取的bi的最少的个数x，答案就是i + x取最小值。设最大值是s前缀和是sum查询的就是最少多少个bi可以大于等于s - sum，，这个数据结构不就是权值线段树吗。。 这个1e9得离散化一下，，，，这里题解给的是在线段树上二分。。由于我不知道咋弄，所以就算了，。。
可能数据很水?

代码

#include <cstdio>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;
const int maxn = 1e6+6;
typedef long long ll;
std::vector<ll> vv;
struct NO
{
	ll a,b;
}a[maxn];
bool cmp(NO a, NO b)
{
	return a.a > b.a;
}
struct Node
{
	int num;
	ll sum;
}node[maxn<<2];
int getid(ll x)
{
	return lower_bound(vv.begin(),vv.end(),x) - vv.begin() + 1;
}
void add(int l,int r,int no,int num,int f)
{
	node[no].sum += f * vv[num - 1];
	node[no].num += f;
	if(l == r)
		return;
	int mid = l + r>> 1;
	if(num <= mid)
		add(l,mid,no<<1,num,f);
	else
		add(mid + 1, r, no<<1|1,num,f);
}

int query(int l,int r,int no,ll sum)
{
	
	if(l == r)
	{
		ll num = vv[l - 1];
		int c = (sum + num - 1) / num;
		if(c > node[no].num)
			return -1;
		else
			return c;
	}
	ll s = node[no<<1|1].sum;
	int x = node[no<<1|1].num;
	int mid = l + r >> 1;
	// printf("%lld %lld %d %d %d\n",s,sum,l,r,x);
	if(s >= sum)
	{
		int y = query(mid + 1, r ,no<<1|1,sum);
		return y;
	}
	else
	{
		int y = query(l, mid,no<<1,sum - s);
		// printf("%d\n",y);
		if(y == -1)
			return -1;
		return x + y;
	}
}

struct P
{
	int id;
	ll val;
	bool operator< (const P & a)const
	{
		return a.val > val;
	}
};
int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	ll x = 0;
	for (int i = 1; i <= n; i ++ )
	{
		scanf("%lld",&a[i].a);
	}
	for (int i =1; i <= n; i ++ )
	{
		scanf("%lld",&a[i].b);
		vv.push_back(a[i].b);
	}
	sort(vv.begin(),vv.end());
	vv.erase(unique(vv.begin(),vv.end()),vv.end());
	int k = vv.size();
	x = a[m].a;
	for (int i = m; i < n; i ++ )
		a[i] = a[i + 1];
	sort(a + 1,a + n,cmp);
	for (int i = 1; i < n; i ++ )
	{
		add(1,k,1,getid(a[i].b),1);
	}
	ll maxx =0 ;
	ll sum = x;
	int ans = -1;
	// printf("%lld %lld\n",sum,qq.top().val);
	if(sum < a[1].a)
	{
		ans = query(1,k,1,a[1].a - sum);
	}
	else
		ans = 0;
	for (int i= 1; i < n; i ++ )
	{
		sum += a[i].b;
		maxx = max(maxx,a[i].a - a[i].b);
		add(1,k,1,getid(a[i].b),-1);
		 // printf("1111111111111 %d %d\n",ans,i);
		ll s = a[i + 1].a;
		s = max(maxx,s);
		if(sum >= s)
		{
			if(ans == -1)
				ans = i;
			else
				ans = min(ans,i);
		}
		else
		{
			s -= sum;
			int x = query(1,k,1,s);
			if(x == -1)
				continue;
			if(ans == -1)
				ans = x + i;
			else
				ans = min(ans,x + i);
		}
		
	}
	if(ans != -1)
	{
		printf("%d\n",ans);
	}
	else
		printf("The Soviet Union is doomed\n");
}