小蓝最近正在玩一款 RPG 游戏。
他的角色一共有 N 个可以加攻击力的技能。
其中第 i 个技能首次升级可以提升 Ai 点攻击力,以后每次升级增加的点数都会减少 Bi。
⌈AiBi⌉(上取整)次之后,再升级该技能将不会改变攻击力。
现在小蓝可以总计升级 M 次技能,他可以任意选择升级的技能和次数。
请你计算小蓝最多可以提高多少点攻击力?
输入格式
输入第一行包含两个整数 N 和 M。
以下 N 行每行包含两个整数 Ai 和 Bi。
输出格式
输出一行包含一个整数表示答案。
数据范围
对于 40% 的评测用例,1≤N,M≤1000;
对于 60% 的评测用例,1≤N≤104,1≤M≤107;
对于所有评测用例,1≤N≤105,1≤M≤2×109,1≤Ai,Bi≤106。
输入样例:
3 6
10 5
9 2
8 1
输出样例:
47
思路 :
- 首先贪心的想法就是每次升级必然选择当前收益最高的,枚举m次用堆维护收益最高的技能是O(mlogn)会tle
- 如果每次都选择收益最大的技能i,我们将每次选择得到的攻击力 排成一个序列,这个序列一定是单调递减的。通过序列的单调性萌生了一个二分答案的想法——二分最后一次选择得到的攻击力l
- check 函数自然就是判断选择次数是否 ≥ mid,此时我们需要一个快速得到选择次数的方法,可以简单推导一下:
a i − ( x − 1 ) b i ≥ m i d a_i-(x-1)b_i\geq mid ai−(x−1)bi≥mid(x为选择次数)
x ≤ a i − m i d b i + 1 x \leq \frac{a_i-mid}{b_i}+1 x≤biai−mid+1 - 由此,我们用O(1)求得选择次数,用O(nlog1e6)二分答案,得到最大的最后一次选择得到的攻击力,显然在这种情况下总共提高的攻击力是最大的
- 由于总计升级 M 次技能,因此不是所有得到攻击力等于mid的都被选择,因此,遍历n个,我们设变量res为将所有攻击力大于等于mid的都累加,cnt为所有攻击力大于等于mid的个数,最终答案就是用res(所有大于等于mid)的减去(cnt - m)个mid
#include <iostream>
using namespace std;
typedef long long ll;
const int N = 1e5 + 10;
int n, m;
int a[N], b[N];
bool check(int mid) {
ll res = 0;
for (int i = 0; i < n; ++ i) {
if (a[i] >= mid) {
res += (a[i] - mid) / b[i] + 1;
}
}
return (res >= m);
}
int main() {
scanf("%d%d", &n, &m);
for (int i = 0; i < n; ++ i) scanf("%d%d", &a[i], &b[i]);
int l = 0, r = 1e6;
while (l < r) {
int mid = l + r + 1 >> 1;
if (check(mid)) l = mid;
else r = mid - 1;
}
ll res = 0, cnt = 0;
for (int i = 0; i < n; ++ i) {
if (a[i] >= r) {
int c = (a[i] - r) / b[i] + 1;
cnt += c;
int end = a[i] - (c - 1) * b[i];
res += (ll)(a[i] + end) * c / 2;
}
}
printf("%lld", res - (cnt - m) * r);
}