感觉可以用优先队列,最后找出前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;
}