【STL】next_permutation的实现

本文探讨了STL中的next_permutation函数的两种实现方法:字典序法和康托逆展开。字典序法通过找到特定位置的元素进行交换和反转来生成下一个排列。康托逆展开则是通过逆康托展开的过程来实现排列的转换。内容来源于LeetCode的相关题目及《STL源码剖析》。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

一、使用字典序法

首先,从最尾端开始寻找两个相邻的元素,令第一个元素为*i, 第二个元素为*ii, 且满足 *i < *ii(其中pos[*i] < pos[*ii], pos代表下标)。找到这样一组相邻的元素之后,再从最尾端开始往前检验,找出第一个大于*i的元素,令为 *j, 将i, j,元素对调,再将ii之后的所有元素颠倒顺序,即可求出“下一个”排列组合。


以下解法为题目: leetcode next permutation

class Solution {
public:
    void nextPermutation(vector<int>& nums) {
        int len = nums.size();
        
        // 只有只一个元素或为空
        if(len < 2) return;
        
        // 如果当前元素已经是最大排列时,下一个则是最小排列
        bool flag = false;
        for(int i = 0; i < len-1; ++ i) {
            if(nums[i] < nums[i+1]) {
                flag = true;
                break;
            } 
        }
        if(!flag) {
            reverse(nums.begin(), nums.end());
            return;
        }
        
        // 找到从最尾端开始的满足*i < *ii的相邻元素
        int posx = len-2, posy = len-1;
        for(int i = len-2; i >= 0; i --) {
            if(nums[i] < nums[i+1]) {
                posx = i;
                posy = i + 1;
                break;
            }
        }
        
        // 从最尾端开始找到第一个大于 *i 的元素,并与之对调
        for(int i = len - 1; i >= 0; -- i) {
            if(nums[i] > nums[posx]) {
                swap(nums[posx], nums[i]);
                break;
            }
        }
        
        // 反转ii之后的所有元素
        reverse(nums.begin() + posy, nums.end());
    }
};


二、康托逆展开

对于给定一个n和一个k,求由1-n组成的第k大排列

以下解法的题目来源:leetcode permutation sequence

class Solution {
public:
    int fac(int n) {
        int ans = 1;
        for(int i = 1; i <= n; ++ i) ans *= i;
        return ans;
    }
    
    string getPermutation(int n, int k) {
        
        // 获取当前(n-1)!
        int cnt = fac(n - 1);
        
        // 组成排列的所有元素
        vector<int> permutation;
        for(int i = 1; i <= n; ++ i) permutation.push_back(i);
        
        string ans = "";
        for(int i = n; i >= 1; i --) {
            // idx 为右边比当前i小的元素个数
            int idx = (k-1) / cnt;
            // 置当前元素
            ans += to_string(permutation[idx]);
            // 去除当前元素以避免产生重复数
            permutation.erase(permutation.begin() + idx);
            // 获取当前剩下的第k大排列
            k = k - idx * cnt;
            // 防止 n = 1 的情况,除0异常
            if(i > 1) cnt /= (i-1);
        }
        
        return ans;
    }
};
已知一个排列,可以用康托展开求其序数。假设排列为,这个排列对应的序数为 。其中 表示位于第i位右边比 小的元素的个数, .

上述过程即为康托展开的过程,而给出的代码为逆康托展开


参考:

侯捷,  STL源码剖析


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值