next_permutation和prev_permutation是STL里面十分有用的两个函数,那么今天我们就来说一说这两个函数的用处。
next_permutation的用法
next_permutation: 查找数列的下一个全排列。
prev_permutation: 查找数列的上一个全排列。
For example:
今有一数列 1,2,3,4
它的全排列为
1 2 3 4
1 2 4 3
1 3 2 4
1 3 4 2
1 4 2 3
1 4 3 2
2 1 3 4
2 1 4 3
2 3 1 4
2 3 4 1
2 4 1 3
2 4 3 1
3 1 2 4
3 1 4 2
3 2 1 4
3 2 4 1
3 4 1 2
3 4 2 1
4 1 2 3
4 1 3 2
4 2 1 3
4 2 3 1
4 3 1 2
4 3 2 1
我们可以像用sort一样使用它。
int a[4]={1,2,3,4}; //数列
next_permutation(a,a+4); //寻找下一个排列
for(auto i:a) printf("%d ",i); //输出看看
这时候我们可以看到输出了 1 2 4 3 ,这正是{1,2,3,4}的下一个排列。
next_prtmutation的返回值
next_prtmutation是有返回值的,它的返回值为数组是否还存在下一个排列。
于是我们就可以利用它的返回值写出如下一段代码。
#include<bits/stdc++.h>
using namespace std;
int main(){
int a[]={1,2,3,4}; //数列
do{
for(auto i:a) printf("%d ",i); //输出数列
printf("\n"); //换行
}while(next_permutation(a,a+4)); //当数列还存在下一个排列时,便继续,否则退出。
}
这就可以输出数列的全排列了。
#include<bits/stdc++.h>
using namespace std;
int a[10010],n;
int main(){
scanf("%d",&n);
for(int i=0;i<n;i++) scanf("%d",&a[i]);
sort(a,a+n);
do{
for(int i=0;i<n;i++) printf("%d ",a[i]);
printf("\n");
}while(next_permutation(a,a+n));
}
next_permutation的其他用法
next_permutation还有一些其他的用法。比如它可以像sort一样,加一个cmp函数。
For example:
#include<bits/stdc++.h>
using namespace std;
bool cmp(const int& a,const int& b){ //自定义的cmp函数
return (a>b);
}
int main(){
int a[]={4,3,2,1}; //数列
do{
for(auto i:a) printf("%d ",i); //输出数列
printf("\n"); //换行
}while(next_permutation(a,a+4,cmp)); //当数列还存在下一个排列时,便继续,否则退出。
}
#include<bits/stdc++.h>
using namespace std;
int a[10010],n;
bool cmp(const int& a,const int& b){
return (a>b);
}
int main(){
scanf("%d",&n);
for(int i=0;i<n;i++) scanf("%d",&a[i]);
sort(a,a+n,cmp);
do{
for(int i=0;i<n;i++) printf("%d ",a[i]);
printf("\n");
}while(next_permutation(a,a+n,cmp));
}
prev_permutation的用法
和next_permutation一样的,只不过这是上一个排列。
#include<bits/stdc++.h>
using namespace std;
int main(){
int a[]={4,3,2,1}; //数列
do{
for(auto i:a) printf("%d ",i); //输出数列
printf("\n"); //换行
}while(prev_permutation(a,a+4)); //当数列还存在上一个排列时,便继续,否则退出。
}
prev_permutation的返回值
也是和next_permutation一样的——它的返回值为数组是否还存在上一个排列。
例题
题目描述
人类终于登上了火星的土地并且见到了神秘的火星人。人类和火星人都无法理解对方的语言,但是我们的科学家发明了一种用数字交流的方法。这种交流方法是这样的,首先,火星人把一个非常大的数字告诉人类科学家,科学家破解这个数字的含义后,再把一个很小的数字加到这个大数上面,把结果告诉火星人,作为人类的回答。
火星人用一种非常简单的方式来表示数字――掰手指。火星人只有一只手,但这只手上有成千上万的手指,这些手指排成一列,分别编号为 1,2,3,⋯。火星人的任意两根手指都能随意交换位置,他们就是通过这方法计数的。
一个火星人用一个人类的手演示了如何用手指计数。如果把五根手指――拇指、食指、中指、无名指和小指分别编号为 1,2,3,4 和 5,当它们按正常顺序排列时,形成了 5 位数 12345,当你交换无名指和小指的位置时,会形成 5 位数 12354,当你把五个手指的顺序完全颠倒时,会形成 54321,在所有能够形成的 120 个 5 位数中,12345 最小,它表示 1;12354 第二小,它表示 2;54321 最大,它表示 120。下表展示了只有 3 根手指时能够形成的 6 个 3 位数和它们代表的数字:
三进制数 | 代表的数字 |
---|---|
123 | 1 |
132 | 2 |
213 | 3 |
231 | 4 |
312 | 5 |
321 | 6 |
现在你有幸成为了第一个和火星人交流的地球人。一个火星人会让你看他的手指,科学家会告诉你要加上去的很小的数。你的任务是,把火星人用手指表示的数与科学家告诉你的数相加,并根据相加的结果改变火星人手指的排列顺序。输入数据保证这个结果不会超出火星人手指能表示的范围。
输入格式
共三行。
第一行一个正整数 N,表示火星人手指的数目(1≤N≤10000)。
第二行是一个正整数 M,表示要加上去的小整数(1≤M≤100)。
下一行是 1 到 N 这 N 个整数的一个排列,用空格隔开,表示火星人手指的排列顺序。
输出格式
N 个整数,表示改变后的火星人手指的排列顺序。每两个相邻的数中间用一个空格分开,不能有多余的空格。
输入输出样例
输入 #1
5 3 1 2 3 4 5
输出 #1
1 2 4 5 3
说明/提示
对于 30% 的数据,N≤15。
对于 60% 的数据,N≤50。
对于 100% 的数据,N≤10000。
题解
这道题用next_permutation是2分钟就能搞定的题目啊!
我们只需要重复m次用next_permutation再输出就好了。
#include<bits/stdc++.h>
using namespace std;
int a[10010],n,m;
int main(){
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>a[i]; //输入原始数列
while(m--) next_permutation(a+1,a+1+n); //重复m次寻找下一个排列
for(int i=1;i<=n;i++) cout<<a[i]<<" "; //输出答案
}
总结
next_permutation和prev_permutation在有些题中还是很好用的,就比如上面那道题,不需要自己手写,9行代码就能搞定了。