[HDU6757]Hunting Monsters

题目

传送门 to HDU

题意概要
有 $n$ 个项目，第 $i$ 个项目会花费 $a_i$ 元，得到 $b_i$ 元的利润。不保证 $a_i⩽b_i$ 。显然钱的总量在任意时刻都必须是自然数。

请问最开始至少有多少钱时，能够完成至少 $k$ 个项目。对于 $k=1,2,3,\dots ,n$，都请求出结果。

数据范围与提示
数据组数 $T⩽600$ 满足 $n⩽3\times 10^5$ 且 $\sum n⩽2.5\times 10^6$ 。

你花钱大手大脚的，保证 $1⩽a_i,b_i⩽10^9$ 。

思路

基本思路

如果所有项目都必须完成，那么完成任务的顺序应当是：

• 先完成 $a_i⩽b_i$ 的，按照 $a_i$ 不降的顺序。
• 再完成 $a_i>b_i$ 的，按照 $b_i$ 不升的顺序。

用 $\text{exchange argument}$ 易证后者。前者是显然的。

暴力做法

若枚举 $a_i⩽b_i$ 的前缀，问题转化为，最开始有 $w$ 元，能选多少 $a_j>b_j$ 的项目？

考虑 $\mathtt{dp}$，用 $f_i(j)$ 表示选 $j$ 个所花费的最少钱，有转移
f i ( j ) = min ⁡ { f i − 1 ( j ) ,    f i − 1 ( j − 1 ) + a i − b i } f_i(j)=\min\{f_{i-1}(j),\;f_{i-1}(j{-}1){+}a_i{-}b_i\} fi​(j)=min{fi−1​(j),fi−1​(j−1)+ai​−bi​}
后者转移条件是 $f_{i-1}(j-1)+a_i⩽w$，这就是之前我说的凸包类型的贪心。

正确做法

考虑推广 暴力做法。直接求 $f_i(j)$ 为最少的 “启动代价”，则有转移
f i ( j ) ← max ⁡ { a i ,    f i − 1 ( j − 1 ) + a i − b i } f_i(j)\gets\max\{a_i,\;f_{i-1}(j{\rm-}1){\rm+}a_i{\rm-}b_i\} fi​(j)←max{ai​,fi−1​(j−1)+ai​−bi​}
注意这里以 $i$ 为首个项目，故转移顺序是 $b_i$ 从小到大。而且上式是 $i$ 被选的情况，其不被选也是有可能的。

它肯定也类似于凸包，类似于闵可夫斯基和。但含有 $\max$，我们只要把它讨论掉，就能得到结果了。不难发现那等价于 $f_{i-1}(j)$ 和 $b_i$ 的大小比较。然后注意到 $f_{i-1}(j)$ 是单增的（根据其意义是显然的）。

先找到最大 $r$ 使 $f_{i-1}(r)⩽b_i<a_i$，那么 $f_i(x)(x⩽r)$ 都是直接从 $f_{i-1}(x)$ 得来。只需要考虑 $f_i(x)(r<x)$ 的更新，也就是 $f_{i-1}(x)(r⩽x)$ 向后转移。

注意 $f_{i-1}(r)$ 是唯一用 $a_i$ 转移的，那我们可以建 虚点 $f_{i-1}^{\prime }(r)=b_i⩾f_{i-1}(r)$，统一格式。此时是经典的闵可夫斯基和。也就是说，虚点及其后面的实点，始终构成一个凸包。

最重要的是 $b_i$ 是单调不降的。那么 $r$ 只会增大。增大后，将某个实点改成虚点，由于虚点的函数值（即 $y$ 坐标）比原来的实点的函数值更大，所以仍然构成凸包。那么就一直进行闵科夫斯基和就行了。

这样，我们可以 $\mathcal{O}(n\log n)$ 的求出 $f_n(i)$ 数组。

合并答案

最终怎么得到答案呢？只需要搞清楚 $a_i⩽b_i$ 的项目的效果。当最初的钱的数量达到某个值时，就可以多选几个，那就会使得 $earn$ 减 $cost$ 变大。所以就是一个分段函数
$$g(x)=x+ear{n}_i(cos{t}_i⩽x<cos{t}_{i+1})$$

我们现在想知道，由于 $a>b$ 的项目选 $i$ 个，需要 $f_n(i)$ 的 “启动资金”，那么最初需要多少钱，能够在经过 $a⩽b$ 的加持后，满足这一条件呢？也就是求最小的 $x$ 使得 $g(x)⩾f_n(i)$ 嘛。

显然 $g(x)$ 也是单调的。并且我们知道 $n$ 个特殊点的函数值，$g(cos{t}_i)=cos{t}_i+ear{n}_i$，那就可以先二分找到这个范围，然后看看 $cos{t}_i⩽x<cos{t}_{i+1}$ 是否有解就行了。

当然可以由二分改为 $\mathrm{pointer}$ 的方法，反正不会是瓶颈。总时间复杂度 $\mathcal{O}(n\log n)$ 。

代码

由于数据有问题，必须使用 $\mathrm{scanf}$ 进行读入。

#include <cstring>
#include <cstdio>
#include <iostream>
#include <vector>
#include <algorithm>
#include <queue>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/priority_queue.hpp>
using namespace __gnu_pbds;
using namespace std;
typedef long long int_;
# define rep(i,a,b) for(int i=(a); i<=(b); ++i)
# define drep(i,a,b) for(int i=(a); i>=(b); --i)
inline int readint(){
	int x; scanf("%d",&x); return x;
}
inline void writeint(int x){
	if(x < 0) putchar('-'), x = -x;
	if(x > 9) writeint(x/10);
	putchar((x-x/10*10)^48);
}

const int MaxN = 300005;
const int infty = (1<<30)-1;

struct Enemy{
	int a, b;
	Enemy(){ }
	Enemy(int A,int B){
		a = A, b = B;
	}
	bool operator < (const Enemy &t) const {
		return b < t.b;
	}
};
Enemy enemy[MaxN];
int_ need[MaxN]; // may cost a lot
// priority_queue< int,vector<int>,greater<int> > pq;
__gnu_pbds::priority_queue< int,greater<int> > pq;
void fightEnemy(int n){
	int pos = 0; // fixed length
	int_ cur = 0; // (virtual) need[pos]
	pq.push(infty); // boundary
	for(int i=1; i<=n; ++i){
		while(pq.top()+cur <= enemy[i].b){
			cur += pq.top(); pq.pop();
			need[++ pos] = cur;
		}
		if(pq.top() != infty){
			int x = pq.top(); pq.pop();
			pq.push(x-(enemy[i].b-cur));
		}
		cur = enemy[i].b; // new value
		pq.push(enemy[i].a-enemy[i].b); // minkowski
	}
	for(; !pq.empty(); pq.pop())
		need[++ pos] = (cur += pq.top());
}

struct Comrade{
	int a, b;
	Comrade(){ }
	Comrade(int A,int B){
		a = A, b = B;
	}
	bool operator < (const Comrade &t) const {
		return a < t.a;
	}
};
Comrade comrade[MaxN];
int cost[MaxN]; int_ got[MaxN];
void meetComrade(int n){
	for(int i=1; i<=n; ++i)
		if(comrade[i].a > got[i-1]){
			cost[i] = cost[i-1]+comrade[i].a-got[i-1];
			got[i] = comrade[i].b;
		}
		else{
			cost[i] = cost[i-1];
			got[i] = got[i-1]+comrade[i].b-comrade[i].a;
		}
}

const int Mod = 1e9+7;
int ddg[MaxN], ans[MaxN];
int main(){
	for(int T=readint(); T; --T){
		int n = 0, m = 0;
		int tot = readint();
		rep(i,1,tot) ddg[i] = readint();
		for(int i=1,a,b; i<=tot; ++i){
			a = ddg[i], b = readint();
			if(a > b) enemy[++ m] = Enemy(a,b);
			else comrade[++ n] = Comrade(a,b);
		}
		sort(comrade+1,comrade+n+1);
		sort(enemy+1,enemy+m+1);
		meetComrade(n); fightEnemy(m);
		int rnk = 0, pa = 1, pb = 1;
		for(; rnk!=n&&got[rnk+1]<need[pb]; ++rnk);
		while(pa <= n || pb <= m){
			int_ vb = cost[rnk]+need[pb]-got[rnk];
			if(rnk != n && vb > cost[rnk+1])
				vb = cost[rnk+1]; // got >= vb
			int va = cost[pa]; // not check range
			if(pa > n || vb < va){
				ans[pa+pb-1] = vb%Mod, ++ pb;
				for(; got[rnk+1]<need[pb]; ++rnk)
					if(rnk == n) break;
			}
			else ans[pa+pb-1] = va, ++ pa;
		}
		int xyx = 0; // output
		rep(i,1,tot) xyx = (xyx+
			int_(i)*ans[i]%Mod)%Mod;
		writeint(xyx), putchar('\n');
	}
	return 0;
}