leetcode46. 全排列

leetcode46. 全排列

题目描述

链接: leetcode46.

给定一个不含重复数字的数组 nums ，返回其 所有可能的全排列 。你可以 按任意顺序 返回答案。

示例1：

输入：nums = [1,2,3]
输出：[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]

示例2：

输入：nums = [0,1]
输出：[[0,1],[1,0]]

示例3：

输入：nums = [1]
输出：[[1]]

题解

回溯算法

• java解法

public class Permutations46 { 
    List<List<Integer>> res = new LinkedList<>();

    public List<List<Integer>> permute(int[] nums) {
        LinkedList<Integer> track = new LinkedList<>();
        trackBack(nums, track);
        return res;
    }

    private void trackBack(int[] nums, LinkedList<Integer> track) {
        if (track.size() == nums.length) { // 全排列结束条件为当前结果大小等于数组大小
            res.add(new LinkedList<>(track));
            return;
        }

        for (int num : nums) {
            // 排除不合法的选择
            if (track.contains(num)) {
                continue;
            }
            // 做选择
            track.add(num);
            // 进入下一层决策树
            trackBack(nums, track);
            // 取消选择
            track.removeLast();
        }
    }
}

• golang解法

var res [][]int

func permute(nums []int) [][]int {
	trackBack(nums, []int{})
	return res
}

func trackBack(nums []int, track []int)  {
	if len(track) == len(nums) {
		res = append(res, track)
		return
	}
	for _, num := range nums {
		if isContain(track, num) {
			continue
		}
		temp := append([]int{}, track...)
		temp = append(temp, num)
		trackBack(nums, temp)
	}
}

func isContain(nums []int, num int) bool {
	for _, item := range nums {
		if item == num {
			return true
		}
	}
	return false
}