Leetcode 全排列問題不含重複和含重複剪枝情況

不重複,直接這樣遞歸就行 

class Solution {
    List<List<Integer>> res = new ArrayList<>();
    public List<List<Integer>> permute(int[] nums) {
        dfs(nums,0);
        return res;
    }

    public void dfs(int[] nums,int index){
        if(index==nums.length){
            ArrayList<Integer> list = new ArrayList<>();
            for(int num:nums) list.add(num);
            res.add(list);
        }
        for(int i=index;i<nums.length;i++){
            swap(nums,index,i);
            dfs(nums,index+1);
            swap(nums,index,i);
        }
    }
    
    public void swap(int[] nums,int i, int j){
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;
    }
}

含重複,剪枝葉的時候,需要注意,不是簡單的跳過,而是前面出現過,就要跳過。因爲swap後,順序被打亂了

public class Solution {
    List<List<Integer>> res = new ArrayList<>();
    public List<List<Integer>> permuteUnique(int[] nums) {
        Arrays.sort(nums);
        dfs(nums,0);
        return res;
    }

    public void dfs(int[] nums, int index){
        if(index == nums.length){
            ArrayList<Integer> list = new ArrayList<>();
            for(Integer num:nums) list.add(num);
            res.add(list);
            return;
        }
        for(int i=index;i<nums.length;i++){
            if(!isOk(nums,index,i)) continue;
            swap(nums,index,i);
            dfs(nums,index+1);
            swap(nums,index,i);
        }
    }

    public boolean isOk(int[] nums, int start, int end){
        for(int j=start;j<end;j++){
            if(nums[j]==nums[end]) return false;
        }
        return true;
    }

    public void swap(int[] nums,int i, int j){
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;
    }
}

 

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