试题 算法训练 粘木棍（Java)

问题描述：

有N根木棍，需要将其粘贴成M个长木棍，使得最长的和最短的的差距最小。

输入格式：

第一行两个整数N,M。

一行N个整数，表示木棍的长度。

输出格式：

一行一个整数，表示最小的差距

样例输入：

3 2

10 20 40

样例输出：

10

数据规模和约定：

N, M<=7

思路来源：http://t.csdn.cn/EXh8m

思路：

1. 当我们看到问题中提及道最小的长度时，我们就可以想到穷举法了，也就是bfs（宽度优先算法）或者dfs（深度优先算法）了

2. 在dfs算法中，我们通过尽可能的多粘木棍来求出最大最小值，这个值也正式作为初始值即为N-M，你们或许会好奇为啥？其实很简单，这里举个例子：

比如说7个木棍我们要粘贴3个，那么每根木棍就需要粘贴7-3=4次啦，也就是5个木棍粘贴在一起，而这就是我们所说的最大化，现在第一次得到的最大最小的粘贴一起的木棍和剩下原先两根没有粘贴相比求得最终最大最小的木棍，然后比较两最大最小值的差值，然后下一次换一棍木棍从新开始以上方法，并和上一次的最大最小值的差值进行比较看是否需要更新替换最小值，最后便可得出最大最小值的差值的最小值。

3. 结合关键代码说明:

public static int N;
    public static int M;
    public static int[] arr;
    public static boolean[] vit;
    public static int temp = 100000000;

声明和定义成为属性。

arr:用来存储输入的数据包括后面对数据的修改

vit：用来记录存入数组的数据是否已经被使用过

N、M代表题目的N、M

temp：用来记录每一次得出的最大最小值的差值。

if (T == 0){
            int max = 1,  min = 100000;
            for (int i = 0; i < N; i++){
                if (!vit[i]){
                    max = Math.max(max,arr[i]);
                    min = Math.min(min,arr[i]);
                }
            }
            temp = Math.min(temp,max - min);
            return;

因为dfs算法主要的思想是递归算法，因此我们先说说终止条件，以上代码主要是说木棍粘完设定好的参数进行当此的比较，并且获得当前比较的最大最小值的差值，并且记录到temp中与下次的最大最小值的差值进行比较，得出差值中的最小值。

核心代码：

for (int i = 0; i < N; i++){
            if (!vit[i]){
                vit[i] = true;
                for (int j = 0; j < N; j++){
                    if (j != i && !vit[j]){
                        arr[j] += arr[i];
                        dfs(T - 1);
                        arr[j] -= arr[i];
                    }
                }
                vit[i] = false;
            }

首先先进行判断此时的木棍是否已经被用过，若不是则把其标记为是，然后再除掉它自己本身和其他的没有被粘贴的木棍进行粘贴。

递归后面有一个回溯，这是实现木棍与其他木棍粘贴的作用，既是实现每一种的可能都遍历出来

当当前木棍的所有组合都遍历并且演算完成后，再把当前木棍的状态设置会未使用过的状态，然后方便给下一个木棍进行使用。

4. 完整代码：

import java.util.Scanner;

/**
 * @author Qik~
 * @create 2023-01-17 11:53
 */
public class Main {

    public static int N;
    public static int M;
    public static int[] arr;
    public static boolean[] vit;
    public static int temp = 100000000;

    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        N = scan.nextInt();
        M = scan.nextInt();

        arr = new int[N];
        vit = new boolean[N];
        for (int i = 0; i < arr.length; i++){
            arr[i] = scan.nextInt();
        }

        dfs(N - M);
        System.out.println(temp);

    }

    public static void dfs(int T){
        if (T == 0){
            int max = 1,  min = 100000;
            for (int i = 0; i < N; i++){
                if (!vit[i]){
                    max = Math.max(max,arr[i]);
                    min = Math.min(min,arr[i]);
                }
            }
            temp = Math.min(temp,max - min);
            return;
        }
        for (int i = 0; i < N; i++){
            if (!vit[i]){
                vit[i] = true;
                for (int j = 0; j < N; j++){
                    if (j != i && !vit[j]){
                        arr[j] += arr[i];
                        dfs(T - 1);
                        arr[j] -= arr[i];
                    }
                }
                vit[i] = false;
            }

        }
    }


}

5. 测试结果：