目录
- 1.技能升级
- 1.题目描述
- 2.输入格式
- 3.输出格式
- 4.样例输入
- 5.样例输出
- 6.数据范围
- 7.原题链接
- 2.解题思路
- 3.Ac_code
1.技能升级
1.题目描述
小蓝最近正在玩一款 RPG 游戏。
他的角色一共有 NNN 个可以加攻击力的技能。
其中第 iii个技能首次升级可以提升 Ai A_iAi点攻击力,以后每次升级增加的点数都会减少 Bi B_iBi。 ⌈A i B i⌉⌈\frac {Ai} {Bi}⌉⌈BiAi⌉(上取整)次之后,再升级该技能将不会改变攻击力。
现在小蓝可以总计升级 MMM 次技能,他可以任意选择升级的技能和次数。
请你计算小蓝最多可以提高多少点攻击力?
2.输入格式
输入第一行包含两个整数 NNN 和 MMM。
以下 NNN 行每行包含两个整数 Ai A_iAi和 Bi B_iBi。
3.输出格式
输出一行包含一个整数表示答案。
4.样例输入
3 6
10 5
9 2
8 1
5.样例输出
47
6.数据范围
1 ≤ N ≤ 1 05, 1 ≤ M ≤ 2 × 1 09, 1 ≤ A i , B i ≤ 1 06。1≤N≤10^5 ,1≤M≤2×10^9,1≤Ai,Bi≤10^6。1≤N≤105,1≤M≤2×109,1≤Ai,Bi≤106。
7.原题链接
技能升级
2.解题思路
根据题意思考虑,每一个技能的提升点数都会是一个递减的等差序列,首项为 ai a_iai,公差为 bi b_ibi,那么对于第 iii 个序列应为:
a i, a i− b i, a i−2 b i…, a i−⌈ aibi ⌉ b ia_i,a_i-b_i,a_i-2b_i…,a_i-⌈\frac {a_i} {b_i}⌉b_i ai,ai−bi,ai−2bi…,ai−⌈biai⌉bi
出于贪心考虑,我们每次将升级当前最大的 ai a_iai,这样选够 mmm 次一定为正确答案,所以我们可以使用优先队列模拟,这样的复杂度为 O ( m l o g n ) 。O(mlogn)。O(mlogn)。但 mmm 最大值为 2 × 1 09 2×10^92×109,这样会TLE
。
考虑完成所有选择以后,我们选择的情况一定如下图(红色表示已选),即每个技能一定在其头部选择了若干个元素,总共需要选够 mmm 个元素,因为每个数组都是递减的,所以选择的所有元素一定是所有元素的前 mmm 大的元素。那么这就是一个多路归并问题。
什么是多路归并呢,这里顾名思义就是直接将所有序列合起来,以题目样例举例:
技能1
: 101010 555
技能2
: 999 777 555 333 111
技能3
: 888 777 666 555 444 333 222 111
我们将三个序列合并后按从大到小进行排序得到新序列 HHH:
H=[10,9,8,7,7,6,5,5,5,4,3,3,2,1,1]H=[10,9,8,7,7,6,5,5,5,4,3,3,2,1,1] H=[10,9,8,7,7,6,5,5,5,4,3,3,2,1,1]
样例的 mmm 为 6
,那么答案即为最大前6
个元素之和,这就是多路归并的思想,由贪心可知这样的选择一定正确且我们可以通过在题目的要求下完成如此选法。但显然我们如果暴力存储然后排序仍会TLE
,考虑如何去进行优化。
首先思考多路归并的数组对我们有什么作用?答案是我们可以求解得到我们选择的第 mm m 个数是多少,定义 f ( x )f(x)f(x) 为所有数列中大于等于 xxx 的数有多少个。
从 f ( x )f(x)f(x)的定义可知,若某个值 ttt 满足 f ( t ) = mf(t)=mf(t)=m,那么数列中所有大于等于 ttt 的数恰好就是前 mmm 大的数。又可知当 x < yx<yx<y时,一定满足 f ( x ) ≥ f ( y )f(x)≥f(y)f(x)≥f(y),所以当 xxx 满足 f ( x ) ≥ mf(x)≥mf(x)≥m时, x − 1x-1x−1也一定满足,而 x + 1x+1x+1却不一定满足,由此符合二段性,所以我们可以去二分求解得到 ttt,其实就是求满足 f ( x ) ≥ mf(x)≥mf(x)≥m 的最大值。
比如样例中二分求得的 ttt 就是 6
,因为 f ( 6 ) = mf(6)=mf(6)=m,那么我们把所有大于等于6
的数都选上即是答案。
但是已知 xxx,如何求得 f ( x )f(x)f(x)?对于第 iii 个数列,我们可以选择的个数为 ( ai− x ) / bi+ 1(a_i-x)/b_i+1(ai−x)/bi+1,那么对于任意一个 xxx 我们都可以在 O ( n )O(n)O(n)的时间复杂度求得 f ( x )f(x)f(x)。加上我们二分的复杂度为 O ( l o g n )O(logn)O(logn),那么我们可以在 O ( n l o g n )O(nlogn)O(nlogn)的复杂度求解得到 ttt。
得到 ttt 之后就简单了,我们只需要把每个数列中大于等于 ttt 的值都选上即可。这也很好求解,由于是等差数列,那么每个数组可选的个数为 ( ai− t ) / bi+ 1(a_i-t)/b_i+1(ai−t)/bi+1,首项为 ai a_iai,那么末项为 ai− ( c − 1 ) ∗ bi a_i-(c-1)*b_iai−(c−1)∗bi,已知首项末项以及项数,根据等差数列求和公式 S = n2( a1+ an)S=\frac {n}{2}(a_1+a_n)S=2n(a1+an)可以求出每一列我们选的总和。
需要注意的是,我们求解的 ttt 是满足 f ( t ) > = mf(t)>=mf(t)>=m并不一定恰好是 f ( t ) = mf(t)=mf(t)=m,因为原数列中可能有多个 ttt,我们可能会选多,所以还需要记录下我们总共选了多少个数,最后减去多余的 ttt。比如如果样例中存在两个6
,那么我们就会选到7
个数,需要减去一个6
。
时间复杂度: O ( n l o g n )O(nlogn)O(nlogn)
3.Ac_code
#includeusing namespace std;typedef long long LL;typedef unsigned long long uLL;typedef pair<int, int> PII;#define pb(s) push_back(s);#define SZ(s) ((int)s.size());#define ms(s,x) memset(s, x, sizeof(s))#define all(s) s.begin(),s.end()const int inf = 0x3f3f3f3f;const int mod = 1000000007;const int N = 200010;LL n, m;void solve(){cin >> n >> m;std::vector<LL> a(n), b(n);for (int i = 0; i < n; ++i) cin >> a[i] >> b[i];auto check = [&](int x) {LL s = 0;for (int i = 0; i < n; ++i) {if (a[i] >= x) s += (a[i] - x) / b[i] + 1;}return s >= m;};int l = 0, r = 1e6;while (l < r) {int mid = l + r + 1 >> 1;if (check(mid)) l = mid;else r = mid - 1;}LL ans = 0, cnt = 0;for (int i = 0; i < n; ++i) {if (a[i] >= r) {//可取个数LL c = (a[i] - r) / b[i] + 1;//末项LL end = a[i] - (c - 1) * b[i];//求和ans += (a[i] + end) * c / 2;cnt += c;}}ans -= (cnt - m) * r;cout << ans << '\n';}int main(){ios_base :: sync_with_stdio(false);cin.tie(0); cout.tie(0);int t = 1;while (t--){solve();}return 0;}