【LeetCode】46. 全排列+47. 全排列 II

题目链接:46. 全排列
题目:给定一个 没有重复 数字的序列,返回其所有可能的全排列。
样例:

输入: [1,2,3]
输出:
[
  [1,2,3],
  [1,3,2],
  [2,1,3],
  [2,3,1],
  [3,1,2],
  [3,2,1]
]

思路:dfs+回溯,使用一个标记数组记录每个位置的数是否用过,如果没有用过,就把当前这个位置的数添加进list,递归边界就是list的长度和nums的长度相等,这时候记录答案进ans,当添加完一轮答案之后进行回溯,去掉标记,同时移除最后一次添加元素。
代码:

class Solution {
    
    
    public void dfs(int []nums,boolean vis[],List<Integer> list,List<List<Integer>> ans){
    
    
        if(nums.length==list.size()){
    
    
            ans.add(new ArrayList<>(list));//注意这里如果直接添加list添加的是引用
            return;
        }
        for(int i=0;i<nums.length;i++){
    
    
            if(!vis[i]){
    
    
                list.add(nums[i]);
                vis[i]=true;
                dfs(nums,vis,list,ans);
                vis[i]=false; //回溯去标记
                list.remove(list.size()-1);
            }
        }
    }
    public List<List<Integer>> permute(int[] nums) {
    
    
        List<List<Integer>> ans=new ArrayList<>();
        boolean vis[]=new boolean[nums.length];
        for(int i=0;i<nums.length;i++) vis[i]=false;
        dfs(nums,vis,new ArrayList<>(),ans);
        return ans;
    }
}

这道题其实和47. 全排列 II相似,只是后者不允许重复,那么记录答案时候换成set就可以了。

class Solution {
    
    

    public void dfs(int nums[],boolean vis[],List<Integer> list,Set<List<Integer>> ans){
    
    
        if(nums.length==list.size()){
    
    
            ans.add(new ArrayList<>(list));
            return ;
        }
        for(int i=0;i<nums.length;i++){
    
    
            if(!vis[i]){
    
    
                list.add(nums[i]);
                vis[i]=true;
                dfs(nums,vis,list,ans);
                vis[i]=false; //回溯删除标记
                list.remove(list.size()-1);
            }
        }
    }
    public List<List<Integer>> permuteUnique(int[] nums) {
    
    
        List<List<Integer>> ans=new ArrayList<List<Integer>>();
        if(nums.length==1){
    
    
            List<Integer> list=new ArrayList<>();
            for(int num:nums){
    
    
                list.add(num);
            }
            ans.add(list);
            return ans;
        }
        List<Integer> list=new ArrayList<>();
        boolean vis[]=new boolean[nums.length];
        for(int i=0;i<nums.length;i++) vis[i]=false;
        Set<List<Integer>> listSet=new HashSet<>();
        dfs(nums,vis,list,listSet);
        ans.addAll(listSet);
        return ans;
    }
}

类似题目还有:77. 组合+39. 组合总和

猜你喜欢

转载自blog.csdn.net/dl962454/article/details/115022914