全排列c++代码_LeetCode 0047 全排列II,全排列模板 + 0031 0046

【20200918 每日一题】刷题使我快乐。更多LeetCode答案请见我的GitHubzhangyixing1007/leetcode。

47. 全排列 II, 46. 全排列, 31. 下一个排列,三题套路相同,所以一起说了。

三题都可以基于 31. 下一个排列,所以先说这一题。

31. 下一个排列
难度中等665
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须 原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。 1,2,31,3,2 3,2,11,2,3 1,1,51,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;
    }
}

时间空间如图

3de0de52445c083f8c5d74b373740497.png
47 全排列II 用时

有趣的是,同样的代码,可以直接拿去做 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;
    }
}

占用时间空间如图

723e7fc85d216b532b878a358a6fd545.png

我们可以看出同样的代码做47题(有重复值)比46题(无重复做)要快一些。这有可能是因为47去重,跳过了很多步骤(当然也和具体测试用例有关)。


这样做47题,并不是最快的。因为全排列有很多种做法,不一定非要按照“顺序”进行,只要不重不漏即可。官方解答就给我们展示了一种非常快的解法。

86984d9cbefd8f2f118c74fdb8410232.png
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)
// 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

(如果有空,我再来总结全排列的其他写法。

以上。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值