题目

46. 全排列

题解

简单的回溯模板

代码

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);

    }
}

参考

  1. 回溯算法
  2. 全排列
  3. 回溯算法 + 位掩码(Python 代码、Java 代码)