STL中next_permutation函数快速上手

用法实验

next_permutation()函数功能是输出所有比当前排列大的排列，顺序是从小到大。

例如当前是213，下一个排列就是231.这里是输出0123的全排列实验。

#include<algorithm>
#include<iostream>
using namespace std;

int main()
{
int a[4]={0,1,2};
do{
cout<<a[0]<<a[1]<<a[2]<<endl;
}while(next_permutation(a,a+3));
cout<<a[0]<<a[1]<<a[2]<<endl;
return 0;
}
- 最后一个排列没有下一个排列，用next_permutation会返回false，但是使用了这个方法后，序列会变成字典序列的第一个，如210变成012。
- 当然还有一个prev_permutation()函数功能是输出所有比当前排列小的排列，顺序是从大到小。

具体案例

题目描述

这是一道简单的小学生算术题，问题是这样的：给你四个整型数A,B,C,D，问其中的某一个数字能否通过其他三个数的加减乘除运算得到。每个数只能用一次。如果可以，则输出Yes，否则输出No。比如四个数为1,2,3,6.而1 + 2 + 3 = 6。所以输出Yes。如果四个数为2,2,3,9，则任意的三个数的运算，都不可能让等式成立。所以输出No。提示：在计算机的整型数相除中，5 / 2 = 2,而我们问题中的算术，5 / 2 = 2.5。能帮你的只有这么多了

输入

第一行输入一个整数t，表示有t组测试数据。每组数据，输入四个整数，A,B,C,D。其中0 <= A,B,C,D <= 1000.

输出

对于每组测试数据，按题意输出Yes或者No。

样例输入

2

1 2 3 6

2 2 3 9

样例输出

Yes

No

题解

#include<cstdio>
#include<algorithm>
#include<iostream>
using namespace std;

double a[5];
bool ok;
void dfs(int step,double ans){
if(ok) return;
if(step>2){
if(ans==a[3]) ok=true;
return;
}
for(int i=0;i<4;i++){
if(i==0) dfs(step+1,ans+a[step]);
else if(i==1) dfs(step+1,ans-a[step]);
else if(i==2) dfs(step+1,ans*a[step]);
else if(i==3) dfs(step+1,ans/a[step]);
}
}
int main()
{
int T;
scanf("%d",&T);
while(T--){
for(int i=0;i<4;i++) scanf("%lf",&a[i]);
ok=false;
sort(a,a+4);//先排序
do{
dfs(1,a[0]);
if(ok) break;
}while(next_permutation(a,a+4));
printf("%s\n",ok?"Yes":"No");
}
return 0;
}