AcWing 4656. 技能升级

题目描述

小蓝最近正在玩一款 RPG 游戏。
他的角色一共有 $N$ 个可以加攻击力的技能。
其中第 $i$ 个技能首次升级可以提升 $A_i$ 点攻击力，以后每次升级增加的点数都会减少 $B_i$。
现在小蓝可以总计升级 $M$ 次技能，他可以任意选择升级的技能和次数。
请你计算小蓝最多可以提高多少点攻击力？

思路

这道题我们一开始去想肯定是优先队列，但是M的数值过大，足足有2e9，所以不可行。通过观察我们发现如果操作完的数组的每一个数都大于 $x$ 的话，那他也一定大于所有小于x的数，所以考虑二分。
二分的对象就是操作完以后每一个数的底线。

#include <bits/stdc++.h>

using namespace std;

typedef long long LL;
const int N = 1e5 + 10;

int n, m;
int a[N], b[N];

LL Sum(int r, int n, int t)
{
    int l = r - t * (n - 1);
    return (LL)(l + r) * n >> 1;
}

bool check(int mid)
{
    LL ans = 0;
    for (int i = 1; i <= n; i ++ )
    {
        if (a[i] > mid)
        {
            ans += ceil((double)(a[i] - mid) / (double)b[i]);
        }
    }
    
    return ans <= m;
}

int main()
{
    cin >> n >> m;
    for (int i = 1; i <= n; i ++ ) cin >> a[i] >> b[i];
    
    int l = 0, r = 1e6;
    while (l < r)
    {
        int mid = l + r >> 1;
        if (check(mid)) r = mid;
        else l = mid + 1;
    }
    
    LL ans = 0, res;
    for (int i = 1; i <= n; i ++ )
    {
        if (a[i] > l)
        {
            int now = ceil((double)(a[i] - l) / b[i]);
            res += now;
            ans += Sum(a[i], now, b[i]);
        }
    }
    
    cout << ans + (LL)l * (m - res) << '\n'; // 因为我们二分的是大于x，所以这里的数量不包含等于x，所以加上等于x的数
    return 0;
}

反思

二分的思路没有思考到，也被惯性思维给束缚住了，总想着二分答案，之后不能总想着二分答案，多锻炼自己曲线救国的思维。