P2707 Facer帮父亲 \color{green}{\texttt{P2707 Facer帮父亲}} P2707 Facer帮父亲
[Problem] \color{blue}{\texttt{[Problem]}} [Problem]
- n n n 个旅游景区,每个点有两个参数 a i , b i ( 1 ≤ i ≤ n ) a_i,b_i(1 \leq i \leq n) ai,bi(1≤i≤n)。
- 你需要给每个景区定一个门票 p p p,特别地,当第 i i i 个景区门票价格等于 p p p 时,会有 max { a i − b i × p , 0 } \max \left \{a_i - b_i \times p,0 \right \} max{ai−bi×p,0} 个人进去参观。
- 要求门票的总和不能超过一个数 k k k,求最大总收益。
- 1 ≤ n , k , a i , b i ≤ 1 × 1 0 5 ( 1 ≤ i ≤ n ) 1 \leq n,k,a_i,b_i \leq 1 \times 10^5(1 \leq i \leq n) 1≤n,k,ai,bi≤1×105(1≤i≤n)。
[Solution] \color{blue}{\texttt{[Solution]}} [Solution]
对于第 i i i 个景区而言,如果门票为 p p p 时,收益为 p × ( a i − b i × p ) = p × a i − p 2 × b i p \times (a_i - b_i \times p)=p \times a_i-p^2 \times b_i p×(ai−bi×p)=p×ai−p2×bi(我们先假设这里的 a i − b i × p > 0 a_i-b_i \times p>0 ai−bi×p>0,方便我们讨论)。
当 p p p 增加 1 1 1 时,收益会增加:
[ ( p + 1 ) × a i − ( p + 1 ) 2 × b i ] − [ p × a i − p 2 × b i ] [(p+1) \times a_i - (p+1)^2\times b_i] -[p \times a_i-p^2 \times b_i] [(p+1)×ai−(p+1)2×bi]−[p×ai−p2×bi]
经过计算可得上式的值为 a i − b i − 2 × p × b i a_i - b_i - 2 \times p \times b_i ai−bi−2×p×bi。
当门票的总和增加 1 1 1 时,我们只能让一个景区的门票费用加 1 1 1,那么我们加谁呢?很简单,上面那个式子最大的景区。
如何维护这个最大,相信一定难不倒大家,是的,用个堆就可以了。
注意,如果所有景区收益增加量(就是上面那个式子)是负数(意味着再加门票要亏了)的时候,直接结束,因为题目要求是不超过即可。
当然 a i < b i a_i <b_i ai<bi 的景区我们无需讨论,但是为了使代码简单一点,我的代码里没有特判,其实这样是对的。因为这些点的门票只要从 0 0 0 增加到 1 1 1,就一定是亏了,我们会先考虑赚钱的。当考虑到它们的时候,就意味着算法的结束了。
应该要开 long long
哦。
[code] \color{blue}{\texttt{[code]}} [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() 函数就是快读,我写得丑,大家就自己写吧。