2021-06-22

全排列回溯算法的模板

在刷题时常会遇到求全排列等类似的问题，虽然不同的题会有特殊的解法，但是用回溯算法，效率虽低，却很通用。下面介绍介绍一种全排列的模板。
具体参考：

https://leetcode-cn.com/problems/zi-fu-chuan-de-pai-lie-lcof/

输入一个字符串，打印出该字符串中字符的所有排列。
你可以以任意顺序返回这个字符串数组，但里面不能有重复元素。

示例:

输入：s = “abc”
输出：[“abc”,“acb”,“bac”,“bca”,“cab”,“cba”]

	
class Solution {
    public String[] permutation(String s) {
        int len = s.length();
        //这里可以将s和visited数组作为全局变量，这样trackBack方法就会少两个参数
        boolean[] visited = new boolean[len];
        //用set去重
        Set<String> res = new HashSet<>();
        String partString = "";
        trackBack(s,visited,partString,res);
        String[] ans = new String[res.size()];
        res.toArray(ans);
        return ans;
    }
	

    public void trackBack(String s, boolean[] visited, String partString, Set<String> res){
        if(partString.length() == s.length()){
            res.add(partString.toString());
            return;
        }
        for (int i = 0; i < visited.length; i ++){
            if(visited[i]){
               continue;
            }
            visited[i] = true;
            partString += s.charAt(i);
            trackBack(s,visited,partString,res);
            partString = partString.substring(0, partString.length() - 1);
            visited[i] = false;
        }
    }
}	

对于我们所熟知的[数字全排列]问题，套用以上模板也可写出

class Solution {
    public List<List<Integer>> permute(int[] nums) {
        List<List<Integer>> list = new ArrayList<>();
        List<Integer> arr = new ArrayList<>();
        boolean[] visited = new boolean[nums.length];
       // Arrays.fill(visited,false);
        trackBack(nums, visited, arr, list);
        return list;
    }

    public void trackBack(int[] nums,boolean[] visited,List<Integer>arr,List<List<Integer>> list){
        if(arr.size()==nums.length){
            list.add(new ArrayList(arr));
            return;
        }
        for(int i=0;i<visited.length;i++){
            if(visited[i] )
                continue;
            visited[i] = true;
            arr.add(nums[i]);
            trackBack(nums, visited, arr, list);
            arr.remove(arr.size()-1);
            visited[i] = false;
        }
    }
}