Acwing-4656. 技能升级

感觉可以用优先队列，最后找出前M大的数（多路归并排序的思路），但是这种方式超时了（呜呜呜呜），此种方式时间复杂度为O(mlogn)【选m次，每次的时间复杂度是logn】

注意：pair先按first比较，如果相等，再按照second比较。pair作为priority_queue元素时，使用默认less进行降序排列。

#include <iostream>
#include <algorithm>
#include <queue>

using namespace std;

typedef long long LL;

const int N = 100010;

int n, m;
int a[N], b[N];
priority_queue<pair<int, int>> q;

int main()
{
    scanf("%d%d", &n, &m);
    
    for (int i = 0; i < n; ++ i)
    {
        scanf("%d%d", &a[i], &b[i]);
        q.push({a[i], i});
    }
    
    LL res = 0;
    for (int i = 0; i < m; ++ i)
    {
        auto t = q.top();
//        printf("t.first = %d, t.second = %d\n", t.first, t.second); 
        
        q.pop();
        res += t.first;
        q.push({t.first - b[t.second], t.second});
    }
    
    printf("%lld\n", res);
    return 0;
}

我们发现这个题中m很大，n很小，所以我们的时间复杂度里不能跟m相关，或者跟m相关的话也要取到log级别，我们发现这个题想求前m个数的总和的话，我们可以换一种思路，可以考虑一下从大往小去数，排在第m位的数值是多少，这个数值是可以通过二分来求出来，那么从大到小数排名第m个的数，如果是x的话，怎么把x二分出来呢？

如果x排在第m位的话，相当于在原序列（指的是从大到小排好序的序列）当中，大于等于x的个数，应该是大于等于m个，有可能等于，有可能大于，因为x可能有多个，而且x是满足这个要求的最大的一个数，也就是说大于等于x + 1的数的个数 小于m个

所以问题转化成，能不能快速的求出x，并且能不能快速的计算出前面的总和，如果都可以的话，那么这道题就可以做了。

① 能不能快速的求出第m项的值x？首先我们发现上面的性质是有二段性的。

假设排名第m的数是t，那么当x <= t时，大于等于x的个数是不是一定大于等于m个？答案是肯定的。所以我们发现x可以用二分来找。

二分的时候我们要判断一下某个性质是否成立，相当于问我们能不能快速求出来大于等于x的个数有多少个，我们发现可以用等差数列的公式来求出所要的个数，即（a - x) / b + 1。

② 如何求出前m数的总和？（第一步只是把x求出来了）

我们可以先求出来>= x的总和是多少，但是>=x的数可能比m多一些（比如多k个），再把多的k个数减去就可以了。

我们刚刚用公式求出来>=x的个数有多少，那么数的个数有了，首项也有，末项就可以求出来了，首项有，末项有，公差有，根据等差数列求和公式，就可以求出前面等差数列的总和了，所以我们发现总和也可以用一个公式求出来。

所以，我们二分完之后求总和的话，再扫描一遍整个序列，用O(n)的时间就可以求出来了，总和也求出来了，那么这个题就可以做了。

注意：排名第m个的数有可能是0，因为当m大于所有序列总长度的时候，后面会存在0的情况

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

typedef long long LL;

const int N = 100010;

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

bool check(int x)
{
    LL res = 0;
    for (int i = 0; i < n; ++ i)
        if (a[i] >= x)
            res += (a[i] - x) / 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;
            int end = a[i] - (c - 1) * b[i];
            cnt += c;
            res += c * (LL)(a[i] + end) / 2;
        }
        
    }
    
    printf("%lld\n", res - (cnt - m) * r);
    return 0;
}