装载问题java_回溯算法之装载问题（Java实现）

问题描述

用回溯法编写一个递归程序解决如下装载问题：

有 n 个集装箱要装上 2 艘载重分别为 c1 和 c2的轮船，其中集装箱 i 的

重量为 wi(1≤ i ≤ n)，且∑ 𝑤𝑖 ≤ 𝑐1 + 𝑐2 。

问是否有一个合理的装载方案可以将这 n 个集装箱装上这 2 艘轮船？如果有，请给出装载方案。

示例

当 n=3，c1=c2=50，且 w=[10,40,40]时，可以将集装箱 1 和 2 装到第一艘轮船上，集装箱3装到第二艘轮船上；

如果 w=[20,40,40]时，无法将这 3 个集装箱都装上轮船。

Java代码

import java.util.ArrayList;

import java.util.List;

import java.util.Scanner;

public class Load {

public static int weight1; //记录第一艘船的载重能力

public static int weight2; //记录第二艘船的载重能力

public static int sum1 = 0,sum2 = 0; //分别代表此时第一艘船的载重和所有集装箱的总重量

public static int[] arr; //记录集装箱的重量

public static int n; //集装箱的个数

public static List list = new ArrayList(); //第一艘船的集装箱的装载的重量

public static void main(String[] args) {

Scanner scanner = new Scanner(System.in);

n = scanner.nextInt();

arr = new int[n];

weight1 = scanner.nextInt();

weight2 = scanner.nextInt();

for (int i = 0; i < n; i++) {

arr[i] = scanner.nextInt();

sum2 += arr[i];

}

scanner.close();

backtrack(0);

}

public static void backtrack(int i) {

if(sum1 > weight1) { // 如若超载，则回溯

return;

}

if(i == n) {

if(sum2 - sum1 < weight2) {

System.out.println(list);

}

return;

}

sum1 += arr[i];

list.add(arr[i]);

backtrack(i+1);

sum1 -= arr[i];

list.remove(list.size()-1);

backtrack(i+1);

}

}