【洛谷】二分查找问题[NOIP2015]聪明的质检员 做题笔记

[NOIP2011 提高组] 聪明的质监员

题目链接

[NOIP2011 提高组] 聪明的质监员 - 洛谷

题目描述

小T 是一名质量监督员，最近负责检验一批矿产的质量。这批矿产共有 $n$ 个矿石，从 $1$ 到 $n$ 逐一编号，每个矿石都有自己的重量 $w_i$ 以及价值 $v_i$ 。检验矿产的流程是：

1 、给定 $m$ 个区间 $[l_i,r_i]$；

2 、选出一个参数 $W$；

3 、对于一个区间 $[l_i,r_i]$，计算矿石在这个区间上的检验值 $y_i$：

$$y_i=\sum _{j=l_i}^{r_i}[w_j\ge W]\times \sum _{j=l_i}^{r_i}[w_j\ge W]v_j$$

其中 $j$ 为矿石编号。

这批矿产的检验结果 $y$ 为各个区间的检验值之和。即：$\sum _{i=1}^m{y}_i$

若这批矿产的检验结果与所给标准值 $s$ 相差太多，就需要再去检验另一批矿产。小T 不想费时间去检验另一批矿产，所以他想通过调整参数 $W$ 的值，让检验结果尽可能的靠近标准值 $s$，即使得 $|s-y|$ 最小。请你帮忙求出这个最小值。

输入格式

第一行包含三个整数 $n,m,s$，分别表示矿石的个数、区间的个数和标准值。

接下来的 $n$ 行，每行两个整数，中间用空格隔开，第 $i+1$ 行表示 $i$ 号矿石的重量 $w_i$ 和价值 $v_i$。

接下来的 $m$ 行，表示区间，每行两个整数，中间用空格隔开，第 $i+n+1$ 行表示区间 $[l_i,r_i]$ 的两个端点 $l_i$ 和 $r_i$。注意：不同区间可能重合或相互重叠。

输出格式

一个整数，表示所求的最小值。

样例 #1

样例输入 #1

5 3 15 
1 5 
2 5 
3 5 
4 5 
5 5 
1 5 
2 4 
3 3

样例输出 #1

10

提示

【输入输出样例说明】

当 $W$ 选 $4$ 的时候，三个区间上检验值分别为 $20,5,0$ ，这批矿产的检验结果为 $25$，此时与标准值 $S$ 相差最小为 $10$。

【数据范围】

对于 $10\%$ 的数据，有 $1\le n,m\le 10$；

对于 $30\%$ 的数据，有 $1\le n,m\le 500$ ；

对于 $50\%$ 的数据，有 $1\le n,m\le 5,000$；

对于 $70\%$ 的数据，有 $1\le n,m\le 10,000$ ；

对于 $100\%$ 的数据，有 $1\le n,m\le 200,000$，$0<w_i,v_i\le 10^6$，$0<s\le 10^{12}$，$1\le l_i\le r_i\le n$ 。

题解思路

看到这，一个比较清晰的思路就是二分查找 + 前缀和

题解代码

import java.util.Scanner;

public class Main {

    static int num, m;
    static int[] w, v, left, right;
    static long[] sumOfNum, sumOfValue;
    //最终求的最小值
    static long min, sy;
    static long y, standard;

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        // 矿石的个数
        num = sc.nextInt();
        // 区间的个数
        m = sc.nextInt();
        // 标准值 s
        standard = sc.nextLong();
        // 重量 w 和 价值 v
        w = new int[num + 1];
        v = new int[num + 1];
        //定义二分查找的左右边界,更新右边界为矿石的最大值
        int l = Integer.MAX_VALUE;
        int r = Integer.MIN_VALUE;
        for (int i = 1; i <= num; i++) {
            w[i] = sc.nextInt();
            v[i] = sc.nextInt();
            l = Math.min(l,w[i]);
            r = Math.max(r, w[i]);
        }
        // 检验区间
        left = new int[m + 1];
        right = new int[m + 1];
        for (int i = 1; i <= m; i++) {
            left[i] = sc.nextInt();
            right[i] = sc.nextInt();
        }
        l = l - 1;
        r = r + 2;
        min = Long.MAX_VALUE;
        while (l <= r) {
            int mid = l + (r - l) / 2;
            if (check(mid)) {
                //如果 y > s, 就减少满足条件的数量, 就需要增加 w
                l = mid + 1;
            } else {
                r = mid - 1;
            }
            min = Math.min(min, sy);
        }
        System.out.println(min);
        sc.close();
    }

    static boolean check(int mid) {
        // 前缀和数组，记录符合要求的个数和 符合要求的价值和
        sumOfNum = new long[num + 1];
        sumOfValue = new long[num + 1];
        for (int i = 1; i <= num; i++) {
            if (w[i] > mid) {
                sumOfNum[i] = sumOfNum[i - 1] + 1;
                sumOfValue[i] = sumOfValue[i - 1] + v[i];
            } else {
                sumOfNum[i] = sumOfNum[i - 1];
                sumOfValue[i] = sumOfValue[i - 1];
            }
        }
        // 求 y
        y = 0;
        for (int i = 1; i <= m; i++) {
            y += (sumOfNum[right[i]] - sumOfNum[left[i] - 1]) * (sumOfValue[right[i]] - sumOfValue[left[i] - 1]);
        }
        sy = Math.abs(standard - y);
        return y > standard;
    }
}

不过这段代码耗时有点高，等学了其他算法之后再尝试提高运行效率吧。