洛谷 P1182 数列分段 Section II （Java）题解

题目Link

来源洛谷该题解内名为“thoq”的关于二分看法：

• 个人认为，二分答案由“二分”和“答案”组成。宏观上，二分，是对问题状态空间的一种遍历方式，主要作用是降低时间复杂度；答案，即枚举答案，则是一种解题方法，主要作用是化求解为判定，降低难度。
  具体来讲：
• 答案。在此题中，我们首先要确定答案的范围。显然，答案最小可能是数列中的最大值（记为 l ），最大可能是所有数的和（记为 r）。其次，我们发现，给出一个解，判定这个解是否合法，在此题中显得简单。这启示我们用枚举答案再判定的方法解决此题。
• 二分。二分建立在答案的基础上。此题中答案的排布，是有一定规律的，即，存在一个分界点 p，使得 [l,p)中的数都是非法解，[p,r]中的数都是可行解。而 p 即为所求的最小解。这种性质启示我们用一种特定的方法来枚举答案，于是有了二分。

---

分析

• 分段且每段连续；在数与数之间插上挡板
• 分的 m 段和的最大值最小：分段的方法有 x 种；每一种分段方法都有一个最大值 max；在 x个 max 中找到最小的 max。

样例分析

• n = 5; m = 3;
• 4 （1） 2 （2） 4 （3） 5 （4） 1
• 在n个数中最多插 n - 1 块板子
• 分段m段组合的板子放法:
  - （1）（2）、（1）（3）、（1）（4）、（2）（3）、（2）（4）、（3）（4）；
  - 这就是 1 —> n-1 的 组合数排序

解1

• 递归实现组合枚举
• 由于数据深度多大；只AC一个点，其余全TLE。

import java.util.*;

public class Main {
    static int[] arr;
    static int[] ans;
    static boolean[] st;
    static long res = (long) 1e9;
    static int n;
    static int m;
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        m = sc.nextInt();
        arr = new int[n];
        st = new boolean[n];
        ans = new int[m + 1];
        for (int i = 0; i < n; i++) {
            arr[i] = sc.nextInt();
        }
        //最后一段肯定是包含arr[n-1]
        ans[m] = n;
        dfs(1);
        System.out.println(res);

    }
    public static void dfs(int step){
        if (step >= m){
            //Ai <= 1*10^9;一段和最坏的情况下 会>int
            long max = -1;
            //选好的m个分段组合
            for (int i = 0; i < m; i++) {
                long sum = 0;
                for (int j = ans[i]; j < ans[i + 1]; j++) {
                    sum += arr[j];
                }
                //m段和中的最大值
                max = Math.max(sum,max);
            }
            //最大值中的最小值
            res = Math.min(max,res);
            return;
        }
        //组合枚举
        for (int i = 1 ; i < n ; i++) {
            if (!st[i]){
                ans[step] = i;
                st[i] = true;
                dfs(step + 1);
                st[i] = false;
            }
        }
    }
}

---

二分解

分析

• 根据题意： 这个结果的范围是 一组数中的最大值 max 到 这组数的和 sum
• max : 当 m = n 时,每个数字为一组 其中的最大值是max; 当 m < n 时 每一段和的最大值肯定是 >= max ( ”=“：当输入的数有0时或者本身就是分段数下的最大值)；而最大的情况就是 m = 1时; 全部数为一组也就是 sum。
• 样例 ： 4 2 4 5 1
• max = 5 sum = 16
• 所以答案必在 5 - sum 之间
• 1当 n = m = 5时:分段和的最大值就是 5；且只有这一种分段的方法
• 2当 n < m 时; 参照枚举法的分段组合，分段和最大值 >= 5
• 3当 m = 1 时；答案也就是 16

---

• 根据上述分析,枚举出的答案的范围也就随之确定 [max,sum]
• 而写出的代码如何根据枚举出的答案进行判定呢？
• 通过分段数 m 和 分段和最大值是否=枚举出的答案 res 进行判定
• 关键 : 当二分的答案分割的段数越趋近于 m 时，满足题意；数越大分的段越少：数越小分的段越大。

---

AC解

import java.util.*;

public class Main {
    static int[] arr;
    static long res = (long) 1e9;
    static int n;
    static int m;
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        m = sc.nextInt();
        arr = new int[n];
        //Ai < 10 ^ 9 所以max<int 但 sum可能>int,为了很好比较数据统一使用long
        long max = -1;
        long sum = 0;
        for (int i = 0; i < n; i++) {
            arr[i] = sc.nextInt();
            //arr[i] : int -> long 自动强制转换
            max = Math.max(max,arr[i]);
            sum += arr[i];
        }
        long l = max; long r = sum;
        while (l + 1 < r ){
            long mid = (l + r) / 2;
            if ( check(mid) ){
                r = mid;
            }else {
                l = mid;
            }
        }
        //特殊情况 l 可以满足
        if ( check(l) ){
            System.out.println(l);
        }else{
            System.out.println(r);
        }
    }
    public static boolean check(long res){
        int num = 1;
        long cur = 0;
        for (int i = 0; i < n ; i++) {
            //分段和如果 > res,分段 num++，cur = 0->下一分段和
            if (cur + arr[i] > res){
                num ++;
                cur = 0;
            }
            cur += arr[i];
        }
        /*
         当 num <= m 时说明比 res 大的数分的段数更少；需要让 num 趋近于 m;所以 return true; r = mid;让 res 变小
         当 num > m 时比res小的数的 num 都 > m; 不符合题目要求；我们需要让 num 趋近于 m 所以 return fasle；l = mid；让 res 变大
         */
        return num <= m;
    }
}