【20200918 每日一题】刷题使我快乐。更多LeetCode答案请见我的GitHubzhangyixing1007/leetcode。
47. 全排列 II, 46. 全排列, 31. 下一个排列,三题套路相同,所以一起说了。
三题都可以基于 31. 下一个排列,所以先说这一题。
31. 下一个排列
难度中等665
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须 原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。1,2,3
→1,3,2
3,2,1
→1,2,3
1,1,5
→1,5,1
class Solution {
public void nextPermutation(int[] nums) {
if(nums==null||nums.length<2) return;
int i=nums.length-2;
while(i>=0&&nums[i]>=nums[i+1]) i--;
if(i==-1){
for(i=0; i<nums.length/2; i++)
swap(nums,i,nums.length-1-i);
return;
}
int j=nums.length-1;
while(nums[i]>=nums[j]) j--;
swap(nums,i,j);
for(j=i+1; j<nums.length-j+i; j++)
swap(nums,j,nums.length-j+i);
}
void swap(int[] nums, int i, int j){
int t=nums[i]; nums[i]=nums[j]; nums[j]=t;
}
}
在此基础上,我们来做今天的题目。
47. 全排列 II
难度中等420
给定一个可包含重复数字的序列,返回所有不重复的全排列。 示例: 输入: [1,1,2] 输出: [ [1,1,2], [1,2,1], [2,1,1] ]
把上述31题的具体内容封装在 nextPermutation 子函数中,返回值为布尔类型,整个代码看起来就很简洁了。
class Solution {
public List<List<Integer>> permuteUnique(int[] nums) {
ans=new ArrayList<>();
Arrays.sort(nums);
curr=new ArrayList<>();
for(int num:nums) curr.add(num);
ans.add(new ArrayList<>(curr));
while( nextPermutation() ){}
return ans;
}
List<List<Integer>> ans;
List<Integer> curr;
List<Integer> tmp;
boolean nextPermutation(){
int i=curr.size()-2;
while(i>=0&&curr.get(i)>=curr.get(i+1)) i--;
if(i==-1) return false;
int j=curr.size()-1;
while(curr.get(j)<=curr.get(i)) j--;
int t=curr.get(i); curr.set(i,curr.get(j)); curr.set(j,t);
for(j=i+1; j<curr.size()-1-(j-i-1); j++){
t=curr.get(j);
curr.set(j, curr.get(curr.size()-1-(j-i-1)));
curr.set(curr.size()-1-(j-i-1),t);
}
ans.add(new ArrayList<>(curr));
return true;
}
}
时间空间如图
有趣的是,同样的代码,可以直接拿去做 46题。(只需要把函数名从permuteUnique改为permute)。
class Solution {
public List<List<Integer>> permute(int[] nums) {
ans=new ArrayList<>();
Arrays.sort(nums);
curr=new ArrayList<>();
for(int num:nums) curr.add(num);
ans.add(new ArrayList<>(curr));
while( nextPermutation() ){}
return ans;
}
List<List<Integer>> ans;
List<Integer> curr;
List<Integer> tmp;
boolean nextPermutation(){
int i=curr.size()-2;
while(i>=0&&curr.get(i)>=curr.get(i+1)) i--;
if(i==-1) return false;
int j=curr.size()-1;
while(curr.get(j)<=curr.get(i)) j--;
int t=curr.get(i); curr.set(i,curr.get(j)); curr.set(j,t);
for(j=i+1; j<curr.size()-1-(j-i-1); j++){
t=curr.get(j);
curr.set(j, curr.get(curr.size()-1-(j-i-1)));
curr.set(curr.size()-1-(j-i-1),t);
}
ans.add(new ArrayList<>(curr));
return true;
}
}
占用时间空间如图
我们可以看出同样的代码做47题(有重复值)比46题(无重复做)要快一些。这有可能是因为47去重,跳过了很多步骤(当然也和具体测试用例有关)。
这样做47题,并不是最快的。因为全排列有很多种做法,不一定非要按照“顺序”进行,只要不重不漏即可。官方解答就给我们展示了一种非常快的解法。
class Solution {
boolean[] vis;
public List<List<Integer>> permuteUnique(int[] nums) {
List<List<Integer>> ans = new ArrayList<List<Integer>>();
List<Integer> perm = new ArrayList<Integer>();
vis = new boolean[nums.length];
Arrays.sort(nums);
backtrack(nums, ans, 0, perm);
return ans;
}
public void backtrack(int[] nums, List<List<Integer>> ans, int idx, List<Integer> perm) {
if (idx == nums.length) {
ans.add(new ArrayList<Integer>(perm));
return;
}
for (int i = 0; i < nums.length; ++i) {
// 去重原理:右边元素不能早于左边元素被占用
if (vis[i] || (i > 0 && nums[i] == nums[i - 1] && !vis[i - 1])) {
continue;
}
perm.add(nums[i]);
vis[i] = true;
backtrack(nums, ans, idx + 1, perm);
vis[i] = false;
perm.remove(idx);
}
}
}
// 作者:LeetCode-Solution
// 链接:https://leetcode-cn.com/problems/permutations-ii/solution/quan-pai-lie-ii-by-leetcode-solution/
// 来源:力扣(LeetCode)
// 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
(如果有空,我再来总结全排列的其他写法。
以上。