STL next_permutation 算法原理和自行实现

目标

STL中的next_permutation 函数和 prev_permutation 两个函数提供了对于一个特定排列P,求出其后一个排列P+1和前一个排列P-1的功能。

这里我们以next_permutation 为例分析STL中实现的原理,prev_permutation 的原理与之类似,我们在最后给出它们实现上差异的比较

 

问题:

给定一个排列P,求出其后一个排列P+1是什么。

思路

 

按照字典序的定义不难推出,正序,是一组排列中最小的排列,而逆序,则是一组排列中最大的排列。

从字典序的定义出发,我们可以看到,n个元素的排列全排列中,从给定排列P 求解下一排列P+1 ,假设两个排列有前k位是相同的(0<=k<=n),那么我们只需要在后面n-k个元素的排列 P(n-k)中求得下一个排列即可

既然我们关心的是后面 n-k位的排列,那么不妨开开脑洞,从后向前考察排列。

 

首先我们举一个比较极端的例子:排列 1 2 3 4 5

很显然,这是一个正序排列(递增序列),因此这是这几个数字所组成的排列中最小的排列,记为P1.

现在我们要求出P2,P2是 1 2 3 5 4. 我们可以看到,P2的前三位和P1的前三位的排列完全相同,唯一的变化是最后两位颠倒顺序,这一顺序的颠倒有何玄机呢?——使得最后两位从正序的 4 5 变成了逆序的 5 4.

接着求P3.P3是 1 2 4 3 5. 我们看到,最后两位已经是逆序,不可能有字典序更大的排列,因此必须考虑更多的位,在后3个元素中,3 5 4 显然不是逆序,因此一定存在字典序更大的排列方式,我们由此确定了n-k==3

 

我们现在要在 3 5 4 中求得下一个排列,3 5 4 不是一个逆序,是因为 3 后面有元素大于3 。我们要在大于3的数字中选择最小的那个,和3交换,因为这样可以保证得到最小的首位元素。对于这个例子,我们选择将3和4进行交换,而不是3 和 5,这样得到的首位元素是4. 现在我们得到了排列 4 5 3 。

显然,4 5 3 并不是我们想要的下一个排列,下一个排列是 4 3 5. 观察区别,不难看出,首位元素一定是4,但是5 3 这个子排列是一个逆序排列

为什么会是逆序排列?

因为我们寻找的时候就以是不是逆序为分割点,3 恰好是第一个非逆序的元素,而4作为与3 交换的元素,又比3要大,因此交换后得到的 5 3 一定是逆序的排列。

逆序排列没有下一排列,但是将逆序排列反向后,我们就得到了对应的正序排列,而正序排列是当前元素所能形成的最小排列,因此,4 3 5 是4 为首位元素所能形成最小排列,而前3 位没有变化,故我们得到了下一排列P3.

另外,大于3的最小元素,即4 ,也是第一个大于3的元素,因为 5 4 是个逆序排列

 

更一般地,例如对于可重集排列 1 2 3 7 6 5 2 1 

我们首先寻找第一个非逆序元素,这里是3,然后从后向前寻找第一个大于3的元素,这里是5,交换,得到 5 7 6 3 2 1 的子排列,然后反向,即可得到下一排列。如果没有找到第一个非逆序元素,那么说明该排列已经是最大排列。

 

代码实现:

以字符串为例,实现next_permutation,这里的空的for语句主要是为了压行

 1 /*
 2 *算法实现:STL中的next_permutation实现 
 3 */
 4 #include<cstdio>
 5 #include<cstring>
 6 
 7 void inline swap(char *s1,char *s2){
 8     char t=*s1;
 9     *s1=*s2;
10     *s2=t;
11 }
12 /**
13 *反转字符串函数,s,e分别执行字符串的开始和结尾,不能反转中文 
14 **/
15 void reverse(char *s,char* e){
16     for(e--;s<e;s++,e--)swap(s,e);
17 }
18 
19 bool next_permutation(char *start,char *end){
20     char *cur = end-1, *pre=cur-1;
21     while(cur>start && *pre>=*cur)cur--,pre--;
22     if(cur<=start)return false;
23     
24     for(cur=end-1;*cur<=*pre;cur--);//找到逆序中大于*pre的元素的最小元素 
25     swap(cur,pre);
26     reverse(pre+1,end);//将尾部的逆序变成正序 
27     return true;
28 }
29 
30 int main(){
31     char s1[]="01224",s2[]="8000";
32     reverse(s1,s1+strlen(s1));
33     printf("%s\n",s1);
34     int n=strlen(s2);
35     puts("下一个排列:");
36     int cnt=0;
37     do{
38         puts(s2);
39         cnt++;
40     }while(next_permutation(s2,s2+n));
41     printf("%d",cnt);
42 }

 

将上述代码的next_permutation 中的所有有关pre和cur指针内容比较的部分的符号反转,就得到了prev_nextpermutation

代码如下:

 1 bool prev_permutation(char *start,char *end){
 2     char *cur = end-1, *pre=cur-1;
 3     while(cur>start && *pre<=*cur)cur--,pre--;//这里符号有变化 
 4     if(cur<=start)return false;
 5     
 6     for(cur=end-1;*cur>=*pre;cur--);//这里符号有变化 
 7     swap(cur,pre);
 8     reverse(pre+1,end);
 9     return true;
10 }

 

嗯,终于解决了心中的一大困惑,也希望能帮助各位~晚安~啦啦啦(~ ̄▽ ̄)~*

posted @ 2016-09-28 00:36  仰望高端玩家的小清新  阅读(9383)  评论(0编辑  收藏  举报