回溯法-求全排列-Java

原题不长这样,但我想求一下全排列。输入输出示例如下:

主要采用回溯法,第一次使用,见注释吧。勉强能看明白,说不出来,方法太巧了。

import java.util.*;
public class Test {
    
    public static void main(String[] args) {
        int[] nums = {1,2,3};
        List<List<Integer>> myres = permute(nums);
        System.out.println(myres);
    }
    public static List<List<Integer>> res = new LinkedList<>();

    /* 主函数,输入一组不重复的数字,返回它们的全排列 */
    public static List<List<Integer>> permute(int[] nums) {
    // 记录「路径」
        LinkedList<Integer> track = new LinkedList<>();
        backtrack(nums, track);
        return res;
    }

    // 路径:记录在 track 中
    // 选择列表:nums 中不存在于 track 的那些元素
    // 结束条件:nums 中的元素全都在 track 中出现
    static void  backtrack(int[] nums, LinkedList<Integer> track) {
    // 触发结束条件
        if (track.size() == nums.length) {
            res.add(new LinkedList(track));
            return;
        }
    
        for (int i = 0; i < nums.length; i++) {
            // 排除不合法的选择
            if (track.contains(nums[i]))
                continue;
            // 做选择
            track.add(nums[i]);
            // 进入下一层决策树
            backtrack(nums, track);
            // 取消选择
            track.removeLast();
        }
    }
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章