洛谷 P1618 & STL全排列方法
今天在做洛谷上暴力专项训练,在解P1618题目的过程中发现了STL中全排列在解题中的妙用,特在此分享。
首先让我们先来看一下这道题:
P1618三连击(升级版)
题目描述
将 1, 2,…,9 共 9 个数分成三组,分别组成三个三位数,且使这三个三位数的比例是 A:B:C,试求出所有满足条件的三个三位数,若无解,输出 No!!!。
输入格式
三个数,A,B,C。
输出格式
若干行,每行 3 个数字。按照每行第一个数字升序排列。
输入样例
1 2 3
输出样例
192 384 576
219 438 657
273 546 819
327 654 981
说明/提示
保证 A<B<C。
很简单的一道暴力题,相信各位看完题目都有自己的想法了,我刚看完题目第一反应就是暴力全遍历一遍,就写了9层for循环,然后对每层循环加限制条件,打了七八十行代码又调试了半天才A掉,过题后看了下其他人的题解,发现有人用STL中的next_permutation方法很简洁的就过了。
那么让我们来看一下这个全排列方法的介绍与用法:
next_permutation全排序方法介绍
template<class Iterator>
bool next_permutation (Iterator first, Iterator last);
template<class Iterator, class Compare>
bool next_permutation (Iterator first, Iterator last, Compare comp);
next_permutation方法原型主要有以上两种,此方法对一个可以遍历的集合按照一定的排列顺序排列到下一个排列,如果存在这样的下一个排列,执行并返回true,否则返回false,需要注意的是此方法是改变当前集合,而不是返回一个新集合。
有点不明白?没关系,举一个简单的例子你就理解了,字符串"abc",在字典序中全排列为:
“abc”,“acb”,“bac”,“bca”,“cab”,“cba”
那么我们通过执行以下代码:
string s = "bac";
next_permutation(s.begin(), s.end());
cout << s;
就会输出 “bca”
如果我们想输出“abc”完整的全排序序列,我们可以这样做:
string s = "abc";
do {
cout << s << endl;
} while (next_permutation(s.begin(), s.end()));
输出结果为:
abc
acb
bac
bca
cab
cba
是不是用起来很方便,代码也很简洁,那么让我们看看刚才那道题如何用next_permutation方法解决:
P1618三连击(升级版)全排列方法AC代码
话不多说,直接放代码,一目了然。
#include<bits/stdc++.h>
using namespace std;
int main()
{
int A,B,C;
cin >> A >> B >> C;
int flag=0;
//flag标记是否有解
int num[9]={1,2,3,4,5,6,7,8,9};
do{
int n1=num[0]*100+num[1]*10+num[2];
int n2=num[3]*100+num[4]*10+num[5];
int n3=num[6]*100+num[7]*10+num[8];
//直接按顺序分配数字
if(n1*B==n2*A&&n1*C==n3*A){
flag=1;
cout << n1 << " " << n2 << " " << n3 << endl;
}
}while(next_permutation(num,num+9));
if(flag==0)
cout << "No!!!" << endl;
return 0;
}
往往简单的题目里藏着许多巧妙的方法,就算是自己AC过的题,也不妨看看别人的题解,或许会有意想不到的收获!