题目
题解
简单的回溯模板
代码
import java.util.*;
public class code46 {
// public static int sum = 0;
// public static List<List<Integer>> permute(int[] nums) {
// List<List<Integer>> res = new ArrayList<>();
// List<Integer> list = new ArrayList<>();
// backtrack(res, list, nums, 0);
// return res;
// }
// public static void backtrack(List<List<Integer>> res, List<Integer> list, int
// nums[], int start) {
// if (sum == nums.length) {
// res.add(new ArrayList<>(list));
// return;
// }
// for (int i = 0; i < nums.length; i++) {
// if (sum <= nums.length && !list.contains(nums[i])) {
// list.add(nums[i]);
// sum += 1;
// backtrack(res, list, nums, i);
// sum -= 1;
// list.remove(list.size() - 1);
// }
// }
// }
public static List<List<Integer>> permute(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
List<Integer> list = new ArrayList<>();
int visit[] = new int[nums.length];
dfs(res, list, nums, visit);
return res;
}
public static void dfs(List<List<Integer>> res, List<Integer> list, int nums[], int visit[]) {
if (list.size() == nums.length) {
res.add(new ArrayList<>(list));
return;
}
for (int i = 0; i < nums.length; i++) {
if (visit[i] == 0) {
visit[i] = 1;
list.add(nums[i]);
dfs(res, list, nums, visit);
visit[i] = 0;
list.remove(list.size() - 1);
}
}
}
public static void main(String[] args) {
int nums1[] = { 1, 2, 3 };
List<List<Integer>> res1 = permute(nums1);
System.out.println(res1);
int nums2[] = { 1, 2, 3, 4 };
List<List<Integer>> res2 = permute(nums2);
System.out.println(res2);
}
}