2020.08.22日常总结——一道数学题

$$\text{P2707 Facer帮父亲}$$

$\text{[Problem]}$

• $n$ 个旅游景区，每个点有两个参数 $a_i,b_i(1\le i\le n)$。
• 你需要给每个景区定一个门票 $p$，特别地，当第 $i$ 个景区门票价格等于 $p$ 时，会有 max ⁡ { a i − b i × p , 0 } \max \left \{a_i - b_i \times p,0 \right \} max{ai​−bi​×p,0} 个人进去参观。
• 要求门票的总和不能超过一个数 $k$，求最大总收益。
• $1\le n,k,a_i,b_i\le 1\times 10^5(1\le i\le n)$。

$\text{[Solution]}$

对于第 $i$ 个景区而言，如果门票为 $p$ 时，收益为 $p\times (a_i-b_i\times p)=p\times a_i-p^2\times b_i$（我们先假设这里的 $a_i-b_i\times p>0$，方便我们讨论）。

当 $p$ 增加 $1$ 时，收益会增加：

$$[(p+1)\times a_i-(p+1{)}^2\times b_i]-[p\times a_i-p^2\times b_i]$$

经过计算可得上式的值为 $a_i-b_i-2\times p\times b_i$。

当门票的总和增加 $1$ 时，我们只能让一个景区的门票费用加 $1$，那么我们加谁呢？很简单，上面那个式子最大的景区。

如何维护这个最大，相信一定难不倒大家，是的，用个堆就可以了。

注意，如果所有景区收益增加量（就是上面那个式子）是负数（意味着再加门票要亏了）的时候，直接结束，因为题目要求是不超过即可。

当然 $a_i<b_i$ 的景区我们无需讨论，但是为了使代码简单一点，我的代码里没有特判，其实这样是对的。因为这些点的门票只要从 $0$ 增加到 $1$，就一定是亏了，我们会先考虑赚钱的。当考虑到它们的时候，就意味着算法的结束了。

应该要开 long long 哦。

$\text{[code]}$

const int N=1e5+100;
struct node{
	int pri,a,b;
	inline long long F(){
		return 1ll*a-b-2ll*pri*b;
	}
	bool operator < (node c) const{
		long long FT=1ll*a-b-2ll*pri*b;
		register long long Fc=c.F();
		return FT<Fc;
	}
}t[N];int n,k;long long ans;
priority_queue<node> q;
int main(){
	n=read();k=read();
	for(int i=1;i<=n;i++){
		t[i].a=read();t[i].b=read();
		t[i].pri=0;q.push(t[i]);
	}
	for(int i=1;i<=k;i++){
		node u=q.top();q.pop();
		if (u.F()<0) break;
		ans+=u.F();//累加增量 
		u.pri++;q.push(u);
	}
	printf("%lld",ans);
	return 0;
}

read() 函数就是快读，我写得丑，大家就自己写吧。