题目
解决数组里有重复元素的情况,就将数组排序,然后记录上次一次选择的元素,并在同层回溯中,不重复选择该元素,保证在每层回溯中,相同元素只会被选择一次。
List<List<Integer>> re = new ArrayList<>(); LinkedList<Integer> path = new LinkedList<>(); public List<List<Integer>> combinationSum2(int[] candidates, int target) { Arrays.sort(candidates); //解决重复[1,2,5,1]会出现[1,2,5],[2,5,1]这种情况需要排序 backtracking(candidates, target, 0, 0); return re; } //使用path.stream().mapToInt(Integer:intValue).sum()时间复杂度会增加 public void backtracking(int[] candidates, int target, int startIndex, int sum){ if (sum == target){ re.add(new ArrayList<>(path)); return; } int last = Integer.MAX_VALUE; for (int i = startIndex; i < candidates.length; i++){ if (last == candidates[i]) continue; if (candidates[i] > target) continue; if (candidates[i] + sum > target) continue; last = candidates[i]; sum += candidates[i]; path.add(candidates[i]); backtracking(candidates, target, i + 1, sum); path.removeLast(); sum -= candidates[i]; //记得回退 } }
题目
这道题是选择一个具有重复元素数组的全排列,一般的回溯模板要有startIndex,因为全排列所以每一个没被选择的元素都有可能被使用,所以设置一个visit数组,使用过的元素不能再次选择。同时为了解决重复元素带来的影响,同样设置一个last来记录上一次选择的元素,同层回溯不能选择相同元素。
List<List<Integer>> re = new ArrayList<>(); LinkedList<Integer> path = new LinkedList<>(); public List<List<Integer>> permuteUnique(int[] nums) { int[] visit = new int[nums.length]; Arrays.sort(nums); backtracking(visit, nums); return re; } public void backtracking(int[] visit, int nums[]){ if (path.size() == nums.length){ re.add(new ArrayList<>(path)); return; } int last = Integer.MAX_VALUE; //记录上一次使用的元素,需要数组有序 for (int i = 0; i < nums.length; i++){ if (nums[i] == last) continue; //同层内的相同元素不能重复使用 if (visit[i] == 0){ last = nums[i]; path.add(nums[i]); visit[i] = 1; backtracking(visit, nums); path.removeLast(); visit[i] = 0; } } }