洛谷多校补题 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);elseadd(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;elsereturn 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);}elseans = 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;elseans = min(ans,i);}else{s -= sum;int x = query(1,k,1,s);if(x == -1)continue;if(ans == -1)ans = x + i;elseans = min(ans,x + i);}}if(ans != -1){printf("%d\n",ans);}elseprintf("The Soviet Union is doomed\n");
}