LeetCode: Permutations II 解題報告

Permutations II

Given a collection of numbers that might contain duplicates, return all possible unique permutations.

For example,
[1,1,2] have the following unique permutations:
[1,1,2], [1,2,1], and [2,1,1].

SOLUTION 1:

還是經典的遞歸模板。需要處理的情況是：我們先把Num排序，然后只能連續地選，這樣就可以避免生成重復的solution.
例子：1 2 3 4 4 4 5 6 7 8
444這個的選法只能:4, 44, 444連續這三種選法

我們用一個visit的數組來記錄哪些是選過的。

public class Solution {
    public List<List<Integer>> permuteUnique(int[] num) {
        List<List<Integer>> ret = new ArrayList<List<Integer>>();
        if (num == null || num.length == 0) {
            return ret;
        }
        
        // For deal with the duplicate solution, we should sort it.
        Arrays.sort(num);
        boolean[] visit = new boolean[num.length];
        
        dfs(num, new ArrayList<Integer>(), ret, visit);
        
        return ret;
    }
    
    public void dfs(int[] num, ArrayList<Integer> path, List<List<Integer>> ret, boolean[] visit) {
        int len = num.length;
        if (path.size() == len) {
            ret.add(new ArrayList<Integer>(path));
            return;
        }
        
        for (int i = 0; i < len; i++) {
            // 只能連續地選，這樣就可以避免生成重復的solution.
            // 例子：1 2 3 4 4 4 5 6 7 8
            // 444這個的選法只能:4, 44, 444連續這三種選法
            if (visit[i] || (i != 0 && visit[i - 1] && num[i] == num[i - 1])) {
                continue;
            }
            
            // 遞歸以及回溯
            visit[i] = true;
            path.add(num[i]);
            dfs(num, path, ret, visit);
            path.remove(path.size() - 1);
            visit[i] = false;
        }
    }
}

 

SOLUTION 2:

用一個pre來記錄選過的值，也可以達到同樣的目的，只取第一個可以取到的位置，后面再取也是一樣的解。用pre記下來，后面就不要再取同樣的值了

// SOLUTION 2:
    //  使用一個pre來記錄。只取第一個可以取的位置
    public void dfs(int[] num, ArrayList<Integer> path, List<List<Integer>> ret, boolean[] visit) {
        int len = num.length;
        if (path.size() == len) {
            ret.add(new ArrayList<Integer>(path));
            return;
        }

        long pre = Long.MIN_VALUE;
        for (int i = 0; i < len; i++) {
            int n = num[i];
            // 只取第一個可取的位置，因為別的位置取到的也沒有區別
            if (visit[i] || pre == n) {
                continue;
            }
            pre = n;
            
            // 遞歸以及回溯
            visit[i] = true;
            path.add(n);
            dfs(num, path, ret, visit);
            path.remove(path.size() - 1);
            visit[i] = false;
        }
    }

 

GITHUB:

https://github.com/yuzhangcmu/LeetCode_algorithm/blob/master/permutation/PermuteUnique.java