算法:回溯六 Permutations II数组全排列II

题目

地址:https://leetcode.com/problems/permutations-ii/
Given a collection of numbers that might contain duplicates, return all possible unique permutations.

Example:

Input: [1,1,2]
Output:
[
  [1,1,2],
  [1,2,1],
  [2,1,1]
]

回溯DFS实现

思路:

  1. 给数组nums排序;
  2. Boolean类型数组used记录每个位置是否已经用过。
  3. 重点条件if (i > 0 && nums[i] == nums[i - 1] && !used[i - 1]),如果不是第一个位置,并且跟上一个位置的值一样,并且还没用过,说明不能再做头部位置。举个反例[1, 1, 2], 如果没有这个条件,则会出现两个结果。这个条件还把[2, 1, 1]给去重,比如第二个1在位置1,因为条件!used[i - 1] == true, 则成功去重。
  4. 正常情况是选择,list.remove(list.size() - 1)语句表示不选择,used[i] = false并重置状态。
package backtracking;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

// https://leetcode.com/problems/permutations-ii/
public class PermutationsII {

  public static void main(String[] args) {
    PermutationsII obj = new PermutationsII();
    int[] input = {1,1,2};
    List<List<Integer>> resultList = obj.permuteUnique(input);
    System.out.println(Arrays.toString(resultList.toArray()));
  }

  public List<List<Integer>> permuteUnique(int[] nums) {
    List<List<Integer>> resultList = new ArrayList<List<Integer>>();
    Arrays.sort(nums);
    boolean[] used = new boolean[nums.length];
    if (nums == null || nums.length == 0) {
      return resultList;
    }
    dfs(nums, resultList, new ArrayList<Integer>(), used);

    return resultList;
  }

  private void dfs(int[] nums, List<List<Integer>> resultList, List<Integer> list, boolean[] used) {
    if (list.size() == nums.length) {
      resultList.add(new ArrayList<Integer>(list));
      return;
    }
    for (int i = 0; i < nums.length; i++) {
      if (used[i]) {
        continue;
      }
      if (i > 0 && nums[i] == nums[i - 1] && !used[i - 1]) {
        continue;
      }
      used[i] = true;
      list.add(nums[i]);
      dfs(nums, resultList, list, used);
      used[i] = false;
      list.remove(list.size() - 1);
    }
  }
}

代码下载

https://github.com/zgpeace/awesome-java-leetcode/blob/master/code/LeetCode/src/backtracking/PermutationsII.java

发布了127 篇原创文章 · 获赞 12 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/zgpeace/article/details/103802369
今日推荐