报名明年4月蓝桥杯软件赛的同学们,如果你是大一零基础,目前懵懂中,不知该怎么办,可以看看本博客系列:备赛20周合集
20周的完整安排请点击:20周计划
每周发1个博客,共20周(读者可以按自己的进度选“正常”和“快进”两种计划)。
每周3次集中答疑,周三、周五、周日晚上,在QQ群上答疑:
第9周: 二分
备赛20周活动已经到第10周了,也快到期末考试阶段了。大家先准备考试,有空做做竞赛题。
二分是蓝桥杯省赛的必考知识点,每年都有。本文的例题大多都是蓝桥杯省赛真题。
(本博主曾写过一篇 二分法、三分法,非常详细地介绍了二分法的原理和例题,请参考。今天这篇博客,介绍了二分的基础内容,以及在蓝桥杯中的应用。)
“二分法”是一种思路简单、编码容易、效率极高、应用广泛的算法。在任何算法竞赛中,二分法都是最常见的知识点之一。
二分法的思想很简单,每次把搜索范围缩小一倍,直到找到答案为止。
设初始范围是[L, R],常见的二分法代码这样写:
while (L < R){ //一直二分,直到区间[L,R]缩小到L=R
int mid = (L + R) / 2; //mid是L、R的中间值
if (check(mid)) R = mid; //答案在左半部分[L,mid],更新R=mid
else L = mid + 1; //答案在右半部分[mid+1, R],更新L=mid+1
}
答案在区间[L, R]中,通过中间值mid缩小搜索范围。这样搜索答案:
(1)如果答案在左半部分,把范围缩小到[L, mid],更新R=mid,然后继续。
(2)如果答案在右半部分,把范围缩小到[mid+1, R],更新L=mid+1,然后继续。
经过多次二分,最后范围缩小到L=R,这就是答案。
二分的效率极高,例如在 10 亿 = 2 30 10亿=2^{30} 10亿=230个数字中查找某个数,前提是这些数已经排序,然后用二分法来找,最多只需找30次。把二分法的计算复杂度记为O(logn)。
以猜数字游戏为例,一个[1, 100]内的数字,猜7次就能猜出来。例如猜数字68:
下面是“猜数游戏”的C++代码实现。二分函数bin_search()操作3个变量:区间左端点L、右端点R、二分的中位数mid。每次把区间缩小一半,把L或R更新为mid;直到L= R为止,即找到了答案所处的位置。
#include
using namespace std;
int a[105];
bool check(int x,int mid){ //二分中的检查函数
return x <= a[mid]; //如果x小于等于中间数,返回true
}
int bin_search(int n, int x){ //在数组a中找数字x,返回位置
int L = 1, R = n; //初始范围[L, R]
while (L < R) {
int mid = (L+R)/2;
if (check(x,mid)) R = mid; //答案在左半部分:[L,mid]
else L = mid+1; //答案在右半部分:[mid+1, R]
}
return a[L]; //返回答案
}
int main(){
int n = 100;
for(int i=1;i<=n;i++) a[i]=i; //赋值,数字1~100
int x = 68; //猜68这个数
cout<<"x="<<bin_search(n,x);
}
Java代码
import java.util.*;
class Main {
static int[] a = new int[105];
static boolean check(int x, int mid) { return x <= a[mid]; }
static int bin_search(int n, int x) {
int L = 1;
int R = n;
while (L < R) {
int mid = (L + R) / 2;
if (check(x, mid)) R = mid;
else L = mid + 1;
}
return a[L];
}
public static void main(String[] args) {
int n = 100;
for (int i = 1; i <= n; i++) a[i] = i;
int x = 68;
System.out.println("x=" + bin_search(n, x));
}
}
Python代码
def check(x, mid): return x <= a[mid]
def bin_search(n, x):
L, R = 1, n
while L < R:
mid = (L+R) // 2
if check(x, mid): R = mid
else: L = mid + 1
return a[L]
n = 100
a = [0] * 105
for i in range(1, n + 1): a[i] = i
x = 68
print("x=", bin_search(n, x))
二分法把长度为n的有序序列上O(n)的查找时间,优化到了O(logn)。
注意二分法的应用条件是:序列是单调有序的,从小到大,或从大到小。在无序的序列上无法二分,如果是乱序的,应该先排序再二分。
如果在乱序序列上只搜一次,不需要用二分法。如果用二分,需要先排序,排序复杂度O(nlogn),再二分是O(logn),排序加二分的总复杂度O(nlogn)。如果用暴力法直接在乱序的n个数里面找,复杂度是O(n)的,比排序加二分快。
但是如果不是搜一个数,而是搜m个数。那么先排序再做m次二分的计算复杂度是O(nlogn+mlogn),而暴力法是O(mn)的,当m很大时,二分法远好于暴力法。
做二分法题目时,需要建模出一个有序的序列,并且答案在这个序列中。编程时,根据题目要求确定区间[L, R]范围,并写一个check()函数来更新L和R。
[L, R]的中间值mid,有多种计算方法:
mid = (L+R)/2
mid = (L+R)>>1
mid = L+(R-L)/2
下表(《算法竞赛》清华大学出版社45页)对比了它们的优缺点。一般情况下,用哪个都行。
二分法的经典应用场景是“最小值最大化(最小值尽量大)”、“最大值最小化(最大值尽量小)”。
(1)最小值最大化
有n个牛棚,分布在一条直线上,有k头牛,给每头牛安排一个牛棚住,k
具体操作是:从第1个点出发,然后逐个检查后面的点,第一个距离≥D的点,就是选中的第2点;然后从第2点出发,再选后面第一个距离第2点≥D的点,这是选中的第3点;…继续下去,直到检查完n个点。这一轮猜测的计算复杂度是O(n)的。
检查完n个点,如果选中的点的数量≥k,说明D猜小了,下次猜大点;如果选中的点的数量
用二分法可以加速猜D的过程。设D的初值是一个极大的数,例如就是所有n点的总长度L。接下来的二分操作和前面的“猜数字游戏”一样,经过O(logL)次,就能确定D。
总计算量:一共O(logL)轮猜测,每一轮O(n),总计算量为O(nlogL)。
(2)最大值最小化
经典的例子是“序列划分”问题。有一个包含n个正整数的序列,把它划分成k个子序列,每个子序列是原数列的一个连续部分,第i个子序列的和为Si。在所有s中,有一个最大值。问如何划分,才能使最大的s最小?这就是“最大值(所有子序列和的最大值)最小化”。
例如序列{2, 2, 3, 4, 5, 1},将其划分成k=3个连续的子序列。下面举例2种分法:{(2, 2, 3)、(4, 5)、(1)},子序列和分别是7、9、1,最大值是9;{(2, 2, 3)、(4)、(5,1)},子序列和是7、4、6,最大值是7。第2种分法比第1种好。
仍然用猜的方法。在一次划分中猜一个x,对任意的Si都有Si ≤ x,也就是说,x是所有Si中的最大值。
如何找到这个x?简单的办法是枚举每一个x,用贪心法每次从左向右尽量多划分元素,Si不能超过x,划分的子序列个数为k个。但是枚举所有的x太耗时了。
用二分法可以加速猜x的过程。用二分法在[max, sum]范围内查找满足条件的x,其中max是序列中最大元素的值,sum是所有元素的和。
蓝桥杯2022初赛题:求阶乘
链接1
链接2
题解:
尾零是2×5相乘得到的,所以只需要计算n!中2和5的因子的数量。又因为n!中2的因子数量远大于5的因子数量,所以只需要计算5的因子数量。
例如25!=25×…×20×…×15×…×10×…×5×…,其中的25、20、15、10、5分别有2、1、1、1、1共6个因子5,所以尾零有6个。
(1)通过30%测试。
简单的方法是检查每个n,计算n!的尾零数量,尾零数量等于k的n就是答案。下面代码第11行的for循环n/5次,对于100%的数据, 1 ≤ K ≤ 1 0 18 1≤K≤10^{18} 1≤K≤1018,由于n比k还大,代码超时。
函数check(n)返回n!的尾零数量,也就是计算n!有多少个因子5,读者可以按自己的思路实现这个函数。下面的check()用了巧妙的方法。以25!为例,对5有贡献的是5、10、15、20、25,即5×1、5×2、5×3、5×4、5×5,共有6个5,其中5个5是25/5得到的,即5×i中的5;还有一个5是循环5次后多了一个5,即i×5的5。再例如100!,尾零的数量包括两部分:100/5=20、20/5=4。
#include
using namespace std;
#define ll long long
ll check(ll n) { //计算n!末尾有多少个0
ll cnt = 0;
while (n) cnt += (n /= 5);
return cnt;
}
int main(){
ll k; cin >> k;
for(ll n=5;;n+=5){ //n是5的倍数,它含有因子5
ll cnt = check(n); //cnt是n!的尾零数量
if(cnt==k){ cout << n; break;}
if(cnt>k) { cout <<-1; break;}
}
return 0;
}
(2)通过100%测试。
本题容易想到可以用二分优化,也就是用二分来猜n。因为n递增时,尾零的数量也是单调递增的,符合二分法的应用条件。
下面讨论代码的计算复杂度。第12行的二分是 O ( l o g E ) O(logE) O(logE),这里 E = 1 0 19 E=10^{19} E=1019。第4行做一次check()差不多是O(1)的。总计算量约为 l o g E = l o g 1 0 19 logE = log10^{19} logE=log1019 < 70次。
C++代码
#include
using namespace std;
#define ll long long
ll check(ll n) { //计算n!末尾有多少个0
ll cnt = 0;
while (n) cnt += (n /= 5);
return cnt;
}
int main() {
ll k; cin >> k;
ll L = 0, R = 1e19; //R的初值为一个极大的数
while (L < R) {
ll mid = (L + R) >> 1;
if (check(mid) >= k) R = mid; // mid!的尾零数量超过了k,说明mid大了
else L = mid + 1; // mid小了
}
if (check(R) == k) cout << R ;
else cout << -1;
return 0;
}
java代码
import java.util.Scanner;
public class Main {
public static long check(long n) {
long cnt = 0;
while (n > 0) {
cnt += n / 5;
n /= 5;
}
return cnt;
}
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
long k = scanner.nextLong();
long L = 0;
long R = (long) Math.pow(10, 19);
while (L < R) {
long mid = (L + R) / 2;
if (check(mid) >= k) R = mid;
else L = mid + 1;
}
if (check(R) == k) System.out.println(R);
else System.out.println(-1);
}
}
python代码
def check(n):
cnt = 0
while n:
cnt += n // 5
n //= 5
return cnt
k = int(input())
L = 0
R = 10**19
while L < R:
mid = (L + R) // 2
if check(mid) >= k: R = mid
else: L = mid + 1
if check(R) == k: print(R)
else: print(-1)
分巧克力: 2017年第八届蓝桥杯省赛
链接1
链接2
题解:
(1)通过50%的测试。
先试试暴力法:把边长从1开始到最大边长d,每个值都试一遍,一直试到刚好够分的最大边长为止。编码:边长初始值d = 1,然后d = 2, 3, 4, …一个个试。
#include
using namespace std;
int h[100010],w[100010];
int n,k;
bool check(int d){ //检查够不够分
int num=0;
for(int i=0;i<n;i++) num += (h[i]/d)*(w[i]/d);
if(num>=k) return true; //够分
else return false; //不够分
}
int main(){
cin >>n>>k;
for(int i=0;i<n;i++) cin>>h[i]>>w[i];
int d=1; //正方形边长
while(1){
if(check(d)) d++; //边长从1开始,一个个地暴力试
else break;
}
cout << d-1;
return 0;
}
暴力法的复杂度:n个长方形,长方形的最大边长d。第16行check()一次的计算量是O(n),需要做d次check(),总复杂度O(n×d),而n和d的最大值是 1 0 5 10^5 105,超时。
(2)通过100%测试
一个个试边长d太慢了,现在使用二分,按前面的“猜数游戏”的方法猜d的取值。暴力法需要做d次check(),现在用二分法,只需要做O(logd)次check()。具体操作是:
第一次:开始时d的范围是1~D,试试中间值D/2,如果这个值大了,就把范围缩小为0 ~ D/2,如果这个值小了,就把范围缩小为D/2 ~ D;
第二次,取新的中间值D/4或3D/4,再试…
…
直到找到合适的值为止。
C++代码
整数的二分法的编码虽然简单,但是很容易出错。左右边界L、R和中间值mid的迭代,由于整数的取整问题,极易出错,导致死循环。下面的代码给出了两种写法,有细微而关键的区别,请仔细领会并彻底理解这两种写法。
#include
using namespace std;
int n,k;
const int N=100010;
int h[N],w[N];
bool check(int d){
int num=0;
for(int i=0;i<n;i++) num += (h[i]/d)*(w[i]/d);
if(num>=k) return true; //够分
else return false; //不够分
}
int main(){
cin >> n >> k;
for(int i=0;i<n;i++) cin>>h[i]>>w[i];
int L=1, R=N; //D的初值是R=100010
//第一种写法:
while(L<R) {
int mid=(L+R+1)>>1; //除2,向右取整
if(check(mid)) L=mid; //新的搜索区间是右半部分,R不变,调整L=mid
else R=mid-1; //新的搜索区间是左半部分,L不变,调整R=mid–1
}
cout << L;
//第二种写法:
/* while(L>1; //除2,向左取整
if(check(mid)) L=mid+1; //新的搜索区间是右半部分,R不变,更新L=mid+1
else R=mid; //新的搜索区间是左半部分,L不变,更新R=mid
}
cout << L-1; */
return 0;
}
java代码
import java.util.Scanner;
public class Main {
static int n, k;
static final int N = 100010;
static int[] h = new int[N];
static int[] w = new int[N];
static boolean check(int d) {
int num = 0;
for (int i = 0; i < n; i++)
num += (h[i] / d) * (w[i] / d);
if (num >= k) return true; // 够分
else return false; // 不够分
}
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
n = scanner.nextInt();
k = scanner.nextInt();
for (int i = 0; i < n; i++) {
h[i] = scanner.nextInt();
w[i] = scanner.nextInt();
}
int L = 1, R = N; // D的初值是R=100010
while (L < R) {
int mid = (L + R + 1) >> 1; // 除2,向右取整
if (check(mid)) L = mid; // 新的搜索区间是右半部分,R不变,调整L=mid
else R = mid - 1; // 新的搜索区间是左半部分,L不变,调整R=mid–1
}
System.out.println(L);
scanner.close();
}
}
python代码
def check(d):
global w,h
num = 0
for i in range(len(w)):
num += (w[i]//d) * (h[i]//d)
if num >= k: return True
return False
n,k = map(int,input().split())
w = []
h = []
for i in range(n):
a,b = map(int,input().split())
w.append(a)
h.append(b)
L ,R = 1, 100010
while L < R:
mid = (L+R)//2
if check(mid): L = mid +1
else : R = mid
print(L-1)
跳石头:链接
这是一道二分法应用的套路题:“最小值最大化”。
在n块岩石中移走m个石头,有很多种移动方法。在第i种移动方法中,剩下的石头之间的距离,有一个最小距离ai。在所有移动方法的最小距离ai中,问最大的ai是多少。
在所有可能的最小值中,找最大的那个,就是“最小值最大化”。
如果用暴力法找所有的组合,在n块岩石中选m个石头有n!/m!(n-m)!种组合,情况太多,显然会超时。
可以转换思路,不去找搬走石头的各种组合,而是给出一个距离d,检查能不能搬走m块石头而得到最短距离d。把所有的d都试一遍,肯定能找到一个最短的d。用二分法找这个d即可。
C++代码
下面的代码中,17 ~ 21行用二分法找一个最小距离d。函数check(d)函数检查d这个距离是否合适。
请保证自己能写出代码,确保能完全写出整数二分而不出错。
#include
int len,n,m;
int stone[50005];
bool check(int d){ //检查距离d是否合适
int num=0; //num记录搬走石头的数量
int pos=0; //当前站立的石头
for(int i=1;i<=n;++i)
if(stone[i]-pos < d) num++; //第i块石头可以搬走
else pos = stone[i]; //第i块石头不能搬走
if(num <= m) return true; //要移动的石头比m少,满足条件
else return false; //要移动的石头比m多,不满足条件
}
int main(){
scanf("%d%d%d",&len,&n,&m);
for(int i=1;i<=n;++i) scanf("%d",&stone[i]);
int L=0,R=len,mid;
while(L<R){
mid = L+(R-L)/2;
if(check(mid)) L = mid+1; //满足条件,说明mid小了,调大一点
else R = mid-1; //不满足条件,说明mid大了,调小一点
}
if(!check(L)) L -= 1;
printf("%d\n",L);
return 0;
}
java代码
import java.util.Scanner;
public class Main {
static int len, n, m;
static int[] stone;
static boolean check(int d) {
int num = 0; // num记录搬走石头的数量
int pos = 0; // 当前站立的石头
for (int i = 1; i <= n; ++i)
if (stone[i] - pos < d) num++; // 第i块石头可以搬走
else pos = stone[i]; // 第i块石头不能搬走
if (num <= m) return true; // 要移动的石头比m少,满足条件
else return false; // 要移动的石头比m多,不满足条件
}
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
len = scanner.nextInt();
n = scanner.nextInt();
m = scanner.nextInt();
stone = new int[n + 1];
for (int i = 1; i <= n; ++i) stone[i] = scanner.nextInt();
int L = 0, R = len, mid;
while (L < R) {
mid = L + (R - L) / 2;
if (check(mid)) L = mid + 1; // 满足条件,说明mid小了,调大一点
else R = mid - 1; // 不满足条件,说明mid大了,调小一点
}
if (!check(L)) L -= 1;
System.out.println(L);
scanner.close();
}
}
python代码
len, n, m = map(int, input().split())
stone = [] # 石头i和到起起点的距离
def check(d):
num = 0
pos = 0
for i in range(0,n): # 0到n-1作为石头下标
if (stone[i]-pos < d): # 第i块可以搬走
num += 1
else: pos = stone[i]
if num <= m: return True
else: return False
for i in range(n):
t = int(input())
stone.append(t)
L, R = 0, len
while (L<R):
mid = L+(R-L)//2
if check(mid): L = mid+1
else: R = mid-1
if check(L): print(L)
else: print(L-1)
青蛙过河:2022年第十三届蓝桥杯省赛
链接1 链接2
题解:往返累计2x次相当于单向走2x次。跳跃能力越大,越能保证可以通过2x次。用二分法找到一个最小的满足条件的跳跃能力。设跳跃能力为mid,每次能跳多远就跳多远,用二分法检查mid是否合法。
C++代码
#include
using namespace std;
int h[1000005];
int n,x;
bool check(int mid){
long long sum=0;
for(int i=0;i<mid-1;i++) sum+=h[i]; //mid-1个
for(int i=0, j=mid-1;j<n;i++,j++) { //青蛙位置i,目标位置j
sum += h[j];
if(sum < 2*x) return false;
sum -= h[i];
}
return true;
}
int main(){
cin>>n>>x;
for(int i=1;i<n;i++) cin>>h[i];
int left=0, right=n;
while(left<right){
int mid=(left+right)/2;
if(check(mid)) right = mid;
else left = mid+1;
}
cout<<right;
return 0;
}
java代码
import java.io.*;
public class Main {
static StreamTokenizer reader = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));
static PrintWriter writer = new PrintWriter(new BufferedWriter(new OutputStreamWriter(System.out)));
static int readInt() throws Exception {
reader.nextToken();
return (int)reader.nval;
}
static int N = 1000010;
static int[] sum = new int[N];
static int n, x;
static boolean check(int len) {
for(int i = 1; i + len <= n; i ++) {
if(sum[i + len - 1] - sum[i - 1] < x)
return false;
}
return true;
}
public static void main(String[] args) throws Exception {
n = readInt(); x = readInt();
x *= 2;
for(int i = 1; i < n; i ++) {
int t = readInt();
sum[i] = sum[i - 1] + t;
}
int l = 1, r = n;
while(l < r) {
int mid = l + r >> 1;
if(check(mid)) r = mid;
else l = mid + 1;
}
writer.println(l);
writer.flush();
}
}
python代码
def check(mid):
for i in range(mid, n):
if sum[i] - sum[i-mid] < 2 * x: return False
return True
n, x = map(int, input().split())
h = list(map(int, input().split()))
sum = [0, h[0]]
for i in range(1, len(h)):
sum.append(h[i] + sum[i])
L = 0
R = 100000
while L <= R:
mid = (L + R) // 2
if check(mid): R = mid - 1
else: L = mid + 1
print(L)
技能升级:2022年第十三届省赛
链接1
链接2
下面详细讲解多种方法,它们分别能通过40%、60%、100%的测试。
(1)暴力法:通过40%测试
先试试暴力法,直接模拟题意,升级m次,每次升级时选用最高攻击力的技能,然后更新它的攻击力。
下面用Python写代码。复杂度是多少?第12行升级m次,是O(m)的;第13行选用最大攻击,使用了Python的max()函数,是O(n)的。总复杂度O(mn),只能通过40%的测试。
import math
n,m = map(int,input().split())
a = [] # 存ai
b = [] # 存bi
c = [] # ai/bi
for i in range(n):
a_,b_ = map(int,input().split())
a.append(a_)
b.append(b_)
c.append(math.ceil(a_/b_)) #向上取整
ans = 0
for i in range(m): #一共升级m次
max_num = max(a) #每次升级时,使用最大的攻击
index = a.index(max_num ) #最大攻击对应的序号
a[index] -= b[index] #更新攻击力
if c[index]>0: ans += max_num #累加攻击力
c[index] -= 1
print(ans)
(2)暴力法+优先队列:通过60%测试
上面的代码可以稍做改进。在n个技能中选用最高攻击力,可以使用优先队列,一次操作的复杂度为O(logn)。m次升级,总复杂度O(mlogn),能通过60%的测试。
下面用C++的优先队列priority_queue实现。priority_queue默认是大根堆,用top()读取队列中的最大值。
#include
using namespace std;
const int N = 1e5 + 5;
typedef pair<int, int> PII;
priority_queue<PII> q; //默认是大根堆
PII p;
int main() {
int n, m; cin >> n >> m;
for (int i = 0; i < n; i++) {
int a, b; cin >> a >> b;
q.push(make_pair(a, b));
}
long long ans = 0;
while (m--) { //升级m次
if (q.empty()) break;
p = q.top(); q.pop(); //每次升级时,使用最大的攻击。读队列最大值并删除
ans += p.first; //累加攻击力
p.first -= p.second; //更新攻击力
if (p.first > 0) q.push(p); //重新放进队列
}
cout << ans;
return 0;
}
java代码
import java.util.PriorityQueue;
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
int n = scanner.nextInt();
int m = scanner.nextInt();
PriorityQueue<PII> q = new PriorityQueue<>((a, b) -> b.first - a.first);
for (int i = 0; i < n; i++) {
int a = scanner.nextInt();
int b = scanner.nextInt();
q.add(new PII(a, b));
}
long ans = 0;
while (m > 0 && !q.isEmpty()) {
PII p = q.poll();
ans += p.first;
p.first -= p.second;
if (p.first > 0) q.add(p);
m--;
}
System.out.println(ans);
}
static class PII {
int first;
int second;
PII(int first, int second) {
this.first = first;
this.second = second;
}
}
}
python代码
Python的“暴力法+优先队列”实现。优先队列用堆实现,heapq默认是小根堆,第7行把a取负,把最大值变成了最小值。
from heapq import *
n,m = map(int,input().split())
q = []
ans = 0
for i in range(n):
a,b = map(int,input().split())
heappush(q,(-a,b))
for i in range(m):
p = heappop(q)
a,b = -p[0],p[1]
ans += a
a = max(a - b, 0)
if a != 0: heappush(q,(-a,b))
print(ans)
(3)二分法:通过100%测试
本题的正解是二分法,能通过100%的测试。
本题 m ≤ 2 × 1 0 9 m≤2×10^9 m≤2×109太大,若逐一升级m次必定会超时。但是不能直接对m进行二分,因为需要知道每个技能升级多少次,而这与m无关。
思考升级技能的过程,是每次找攻击力最高的技能。对某个技能,最后一次升级的攻击力,肯定比之前升级的攻击力小,也就是前面的升级都更大。可以设最后一次升级提升的攻击力是mid,对每个技能,若它最后一次能升级mid,那么它前面的升级都更大。所有这样最后能达到mid的技能,它们前面的升级都应该使用。用二分法找到这个mid,另外,升级技能减少的攻击力的过程是一个等差数列,用O(1)次计算即可知道每个技能升级了几次。知道了每个技能升级的次数,就可以计算一共提升了多少攻击力,这就是题目的答案
下面给出Python代码。函数check(mid)找这个mid。第6行,若所有技能升级总次数大于等于m次,说明mid设小了,在19行让L增大,即增加了mid。第7行,若所有技能升级总次数小于m次,说明mid设大了,在20行让R减小,即缩小了mid。
分析代码的复杂度。17~20行,二分 O ( l o g A ) O(logA) O(logA)次,这里A表示 1 ≤ A i ≤ 1 0 6 1≤Ai≤10^6 1≤Ai≤106;每次check()是O(n),二分的总复杂度是O(nlogA)。23行O(n)。代码的总复杂度是O(nlogA) + O(n),通过100%的测试。
c++代码
#include
using namespace std;
typedef long long ll;
ll a[100005],b[100005];
int n, m;
bool check(ll mid) {
ll cnt = 0;
for (int i = 0; i < n; i++) {
if (a[i] < mid) continue;
cnt += (a[i] - mid) / b[i] + 1;
if (cnt >= m) return true;
}
return false;
}
int main() {
cin >> n >> m;
for (int i = 0; i < n; i++) cin >> a[i] >> b[i];
ll L = 1, R = 1000000;
while (L <= R) {
ll mid = (L + R) / 2;
if (check(mid)) L = mid + 1;
else R = mid - 1;
}
ll attack = 0;
ll cnt = m;
for (int i = 0; i < n; i++) {
if (a[i] < R) continue;
ll t = (a[i] - L) / b[i] + 1;
if (a[i] - b[i] * (t - 1) == R) t -1;
attack += (a[i] * 2 - (t - 1) * b[i]) * t / 2;
cnt -= t;
}
cout << attack + cnt * R << endl;
return 0;
}
java代码
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;
public class Main {
static BufferedReader br = new BufferedReader(new InputStreamReader(System.in), 65536);
static StringTokenizer tokenizer = new StringTokenizer("");
static int maxn = (int) (1e5 + 5);
static int[] array = new int[maxn];
static int[] facts = new int[maxn];
static int n, m;
// 二分枚举最后一次攻击力最高能加多少,并且要注意最后的计算不能把等于这个值的直接加在答案里
public static void main(String[] args) {
n = nextInt(); m = nextInt();
for (int i = 0; i < n; i++) {
array[i] = nextInt();
facts[i] = nextInt();
}
long ans = 0L;
// 二分最后一次技能最多提升了多少攻击力
int l = 0, r = (int) 1e6;
while (l < r) {
int mid = l + r + 1 >> 1;
if (check(mid)) l = mid;
else r = mid - 1;
}
int x = l;
for (int i = 0; i < n; i++) {
if (array[i] >= x) {
int cnt = (array[i] - x) / facts[i] + 1;
int last = array[i] - (cnt - 1) * facts[i];
if (last == x) {
cnt--;
last += facts[i];
}
ans += (long) (array[i] + last) * cnt >> 1;
m -= cnt;
}
}
ans += (long) m * x;
System.out.println(ans);
}
// 最后一次加攻击力不能到 x
public static boolean check(int x) {
int cnt = 0;
for (int i = 0; i < n; i++) {
if (array[i] < x) continue;
cnt += (array[i] - x) / facts[i] + 1;
if (cnt >= m) return true;
}
return false;
}
public static String next() {
while (!tokenizer.hasMoreTokens()) {
try {
tokenizer = new StringTokenizer(br.readLine());
} catch (IOException e) {
e.printStackTrace();
}
}
return tokenizer.nextToken();
}
public static int nextInt() { return Integer.parseInt(next()); }
public static long nextLong() { return Long.parseLong(next()); }
}
python代码
def check(mid): #最后一次技能升级,最多能不能到mid
cnt = 0
for i in range(n):
if a[i] < mid: continue #第i个技能的初值还不够mid,不用这个技能
cnt += (a[i] - mid) // b[i] + 1 #第i个技能用掉的次数
if cnt >= m: return True #所有技能升级总次数大于等于m次,说明mid设小了
return False #所有技能升级总次数小于m次,说明mid设大了
n, m = map(int, input().split())
a = [] #存ai
b = [] #存bi
for i in range(n):
a_,b_ = map(int,input().split())
a.append(a_)
b.append(b_)
L,R = 1,1000000 #二分枚举最后一次攻击力最高能加多少
while(L <= R):
mid = (L + R) // 2
if check(mid): L = mid + 1 #增加mid
else: R = mid - 1 #减小mid
attack = 0
cnt = m
for i in range(n):
if a[i] < R: continue
t = (a[i] - L) // b[i] + 1 #第i个技能升级次数
if a[i] - b[i] * (t - 1) == R: t -= 1 #这个技能每次升级刚好等于R,其他技能更好
attack += (a[i] * 2 - (t - 1) * b[i]) * t / 2
cnt -= t
print(int(attack) + cnt * R)