CF1996F Bomb

$\text{[Problem Discription]}$

给定两个长度为 $n$ 的数组 $a_{1\dots n},b_{1\dots n}$。你可以进行 $k$ 次操作，每次操作可以选择一个 $i$，获得 $a_i$ 的收益，然后使 $a_i\leftarrow \max (0,a_i-b_i)$。求 $k$ 次操作后总收益的最大值。

$1\le n\le 2\times 10^5,1\le k,a_i,b_i\le 10^9$。

多组数据但保证 $\sum n\le 2\times 10^5$。

$\text{[Solution]}$

首先考虑如果 $k$ 较小如何暴力处理。

显然，每次操作肯定是选择是最大的 $a_i$。只需要用一个优先队列即可求出答案。

然后再考虑 $k$ 这么大如何处理。

然后你就会发现暴力对正解似乎没有任何帮助。

其实还是有一些的。每次操作选择的是最大的 $a_i$，也就是对于任意的一个 $x$，如果 $\exists i\text{s.t.}{a}_i\le x$，那么我们就一定不会选择小于 $x$ 的 $a_i$ 进行操作。

利用这个性质我们就自然而然地想到了二分。

维护一个关于 $x$ 的函数 $\text{cnt}$，表示使得每个 $a_i$ 通过足够次操作后小于 $x$ 所需要的总操作数。

利用二分，我们可以求出 $x_0$，$x_0$ 是使得 $\text{cnt}(x)\ge k$ 的最大的 $x$。

这个 $x_0$ 就是这么一条线：最优方案下只要一个 $a_i$ 通过若干次操作后比 $x_0$ 小，那么一定不会再被操作。

但是，把每一个 $a_i$ 都进行足够次操作使得其小于 $x_0$ 所需要的总操作数 $\text{cnt}(x_0)$ 却完全可以大于 $k$。题目的要求是最多只能进行 $k$ 次，显然要把多余的 $(\text{cnt}(x_0)-k)$ 次剔除。

不能暴力！

那怎么办？这是最巧妙的地方。多余的 $(\text{cnt}(x_0)-k)$ 次操作的收益一定都是 $x_0$。

为什么？

根据我们的定义：$x_0$ 是使得 $\text{cnt}(x)\ge k$ 的最大的 $x$。如果多余的 $(\text{cnt}(x_0)-k)$ 次操作有某一次操作的收益 $x_1$ 比 $x_0$ 大，那么 $\text{cnt}(x_1)$ 一定也比 $k$ 大，那么 $x_0$ 就不是满足条件的最大的 $x$。

$\text{Code}$

const int N=2e5+100;
typedef long long ll;

int T,n,k,a[N],b[N];

bool check(int mid){
	ll ret=0;//开 ll
	for(int i=1;i<=n;i++)
		if (a[i]>=mid) ret+=(a[i]-mid)/b[i]+1;
	return ret>=k;
}

inline ll calc(int mid){
	ll ans=0,cnt=0;
	for(int i=1;i<=n;i++)
		if (a[i]>=mid){
			int tmp=(a[i]-mid)/b[i]+1;
			cnt+=tmp;
			ans+=1ll*a[i]*tmp-1ll*tmp*(tmp-1)/2*b[i];
		}
	
	return ans+1ll*(k-cnt)*mid;
}

int main(){
	T=read();
	while (T--){
		int l=1,r=0,res=0;
		
		n=read();k=read();
		for(int i=1;i<=n;i++)
			r=max(r,a[i]=read());
		for(int i=1;i<=n;i++)
			b[i]=read();
		
		while (l<=r){
			int mid=(l+r)>>1;
			if (check(mid)){
				l=mid+1;
				res=mid;
			}
			else r=mid-1;
		}//这是一种不需要考虑最终合法解是 l 还是 l+1 还是 r 还是 r-1 的比较方便的二分写法
		
		printf("%lld\n",calc(res));
	}
	
	return 0;
}

read() 是快读函数