LeetCode-47.全排列II(相关话题:回溯算法)

给定一个可包含重复数字的序列,返回所有不重复的全排列。

示例:

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

解题思路:这个题跟上一道题类似,唯一的区别在于要排除重复元素,这里利用了HashMap,存每层已搜索过的元素,在每一层往下搜索之前,查询当前元素是否在当前搜索层的HashMap中,在的话就跳过(这里个人感觉有点像深度优先搜索)。

java代码:

class Solution {
    public List<List<Integer>> permuteUnique(int[] nums) {
        List<List<Integer>> result = new LinkedList<>();
        List<Integer> tmp = new ArrayList<>(nums.length);
        boolean[] flag = new boolean[nums.length];
        dfs(nums, flag, tmp, 0, result);

        return result;
    }

    private void dfs(int[] nums, boolean[] flag, List<Integer> tmp, int cnt, List<List<Integer>> result) {
        if(cnt == nums.length) {
            result.add(tmp);
            return;
        }

        Map<Integer, Integer> f = new HashMap<>();
        for(int i = 0; i < nums.length; i++) {
            if(flag[i] || null != f.get(nums[i]))
                continue;

            f.put(nums[i], i);
            boolean[] flag1 = Arrays.copyOf(flag, flag.length);
            List<Integer> tmp1 = new ArrayList<>(nums.length);
            tmp1.addAll(tmp);

            tmp1.add(nums[i]);
            flag1[i] = true;

            dfs(nums, flag1, tmp1, cnt+1, result);
        }
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_38823568/article/details/81627007
今日推荐