<蓝桥杯软件赛>零基础备赛20周--第10周--二分

报名明年4月蓝桥杯软件赛的同学们,如果你是大一零基础,目前懵懂中,不知该怎么办,可以看看本博客系列:备赛20周合集
20周的完整安排请点击:20周计划
每周发1个博客,共20周(读者可以按自己的进度选“正常”和“快进”两种计划)。
每周3次集中答疑
,周三、周五、周日晚上,在QQ群上答疑:

在这里插入图片描述

文章目录

  • 1. 二分基础
    • 1.1 二分概念
    • 1.2 基本代码
    • 1.3 二分的应用条件
    • 1.4 中值mid的计算方法
  • 2. 经典应用:最小值最大化、最大值最小化
  • 3. 例题
    • 3.1 简单二分:求阶乘
    • 3.2 简单二分:分巧克力
    • 3.3 最小值最大化:跳石头
    • 3.4 简单题:青蛙过河
    • 3.5 中等题:技能升级

第9周:二分

备赛20周活动已经到第10周了,也快到期末考试阶段了。大家先准备考试,有空做做竞赛题。

1. 二分基础

  二分是蓝桥杯省赛的必考知识点,每年都有。本文的例题大多都是蓝桥杯省赛真题。
  (本博主曾写过一篇 二分法、三分法,非常详细地介绍了二分法的原理和例题,请参考。今天这篇博客,介绍了二分的基础内容,以及在蓝桥杯中的应用。)

1.1 二分概念

  “二分法”是一种思路简单、编码容易、效率极高、应用广泛的算法。在任何算法竞赛中,二分法都是最常见的知识点之一。
  二分法的思想很简单,每次把搜索范围缩小一倍,直到找到答案为止。
  设初始范围是[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:
<蓝桥杯软件赛>零基础备赛20周--第10周--二分_第1张图片

1.2 基本代码

  下面是“猜数游戏”的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))

1.3 二分的应用条件

  二分法把长度为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。

1.4 中值mid的计算方法

  [L, R]的中间值mid,有多种计算方法:
    mid = (L+R)/2
    mid = (L+R)>>1
    mid = L+(R-L)/2
  下表(《算法竞赛》清华大学出版社45页)对比了它们的优缺点。一般情况下,用哪个都行。
<蓝桥杯软件赛>零基础备赛20周--第10周--二分_第2张图片

2. 经典应用:最小值最大化、最大值最小化

  二分法的经典应用场景是“最小值最大化(最小值尽量大)”、“最大值最小化(最大值尽量小)”。
(1)最小值最大化
  有n个牛棚,分布在一条直线上,有k头牛,给每头牛安排一个牛棚住,k   “牛棚问题”的求解可以用猜的方法。猜最小距离是D,看能不能在n个点中选k个,使得任意两点之间的距离≥D。如果可以,说明D是一个合法的距离。然后猜新的D,直到找到那个最大的合法的D。
  具体操作是:从第1个点出发,然后逐个检查后面的点,第一个距离≥D的点,就是选中的第2点;然后从第2点出发,再选后面第一个距离第2点≥D的点,这是选中的第3点;…继续下去,直到检查完n个点。这一轮猜测的计算复杂度是O(n)的。
  检查完n个点,如果选中的点的数量≥k,说明D猜小了,下次猜大点;如果选中的点的数量   如何猜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是所有元素的和。

3. 例题

3.1 简单二分:求阶乘

蓝桥杯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} 1K1018,由于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)

3.2 简单二分:分巧克力

分巧克力: 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)

3.3 最小值最大化:跳石头

跳石头:链接
  这是一道二分法应用的套路题:“最小值最大化”。
  在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)

3.4 简单题:青蛙过河

青蛙过河: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)

3.5 中等题:技能升级

技能升级: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 m2×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 1Ai106;每次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)

你可能感兴趣的:(蓝桥杯软件赛零基础备赛20周,蓝桥杯)