搜索算法真是万能的,很多问题都可以“暴力解决”,暴力搜索出答案来。
最简单的搜索莫过于枚举法,枚举整数,子串,但是在枚举过程中,如果能认真分析问题的本质,则可以优化问题。
晚上看了枚举排列算法,发现搜索真是太好用了。
1.生成1~n的排列
题目:输入整数n,按字典序从小到大的顺序输出前n个数的排列。
如当n=3时,(1,2,3)、(1,3,2)、(2,1,3)、(2,3,1)、(3,1,2)、(3,2,1)。
分析这个题目:此题目很明显用递归很简单。先递归调用打印1开头,再打印完所有1开头之后,递归调用打印2开头的,最后递归调用打印3开头的。
伪代码如下:
序列A:(1,2,3) 集合S:{1,2,3}
void print_permutation(序列A,集合S) { if(S为空) 输出序列A else 按照从小到大的顺序依次考虑S的每个元素 print_permutation(在A的末尾添加v后得到的新序列,S-{v}) }
#include<stdio.h>
int A[100];
// 输出1~n的全排列
void print_permutation(int n, int* A, int cur) {
int i, j;
if(cur == n) { // 递归边界
for(i = 0; i < n; i++) printf("%d ", A[i]);
printf("\n");
} else for(i = 1; i <= n; i++) { // 尝试在A[cur]中填各种整数i
int ok = 1;
for(j = 0; j < cur; j++)
if(A[j] == i) ok = 0; // 如果i已经在A[0]~A[cur-1]出现过,则不能再选
if(ok) {
A[cur] = i;
print_permutation(n, A, cur+1); // 递归调用
}
}
}
int main() {
print_permutation(4, A, 0);
return 0;
}
2.生成可重集的排列
题目:输入数组P,并按字典序列输出数组A各元素的所有全排列。
这个题目要注意下面几点:
1.首先将if(A[j]==i)改为if(A[j]==P[i]) ,A[cur]=i改为 A[cur]=P[i]
2.上一题中,不准出现重复元素,但此题目可以出现(1,1,1)的情况,所以可以出现重复情况,解决方法是统计A[0]~A[cur-1]中P[i]的出现次数c1,以及P数组中P[i]的出现次数c2,用c1和c2来确定元素之前有没有出现过,或者出现过了,有没有到达数量上限。只要c1<c2,就递归调用。c1=c2,说明此元素在前面0到cur-1出现过了,就再选择P中下一个元素。
3.如果输入1,1,1,则每一个1都要递归一遍,结果就是27个1,1,1,所以解决方法是:由于P数组已经有序,我们只需检查P的第一个元素和所有“与前一个元素不同的元素”即可,也就是要加上个判断句if(!i || P[i]!=P[i-1]),这样就正确结果了。
#include<stdio.h>
int P[100], A[100];
// 输出数组P中元素的全排列。数组P中可能有重复元素
void print_permutation(int n, int* P, int* A, int cur) {
int i, j;
if(cur == n) {
for(i = 0; i < n; i++) printf("%d ", A[i]);
printf("\n");
} else for(i = 0; i < n; i++) if(!i || P[i] != P[i-1]) {
int c1 = 0, c2 = 0;
for(j = 0; j < cur; j++) if(A[j] == P[i]) c1++;
for(j = 0; j < n; j++) if(P[i] == P[j]) c2++;
if(c1 < c2) {
A[cur] = P[i];
print_permutation(n, P, A, cur+1);
}
}
}
int main() {
int i, n;
scanf("%d", &n);
for(i = 0; i < n; i++)
scanf("%d", &P[i]);
print_permutation(n, P, A, 0);
return 0;
}
3.解答树
排列生成算法可以生成解答树。
如果某个问题的解可以有多个步骤来得到,而每个步骤都有若干选择(这些候选方案集可能会依赖于先前作出的选择),且可以用递归枚举法来实现,则它的工作可以用解答树描述。
解答树上的节点几乎全部来源于最后一两层。和它们相比,上面的节点数可以忽略不计。
4.下一个排序
在C++标准库中有关于全排列的函数next_permutation,生成下一个排序。
用在上题就很简单,代码如下:
#include<cstdio>
#include<algorithm>
using namespace std;
int main() {
int n, p[10];
scanf("%d", &n);
for(int i = 0; i < n; i++) scanf("%d", &p[i]);
sort(p, p+n); // 排序,得到p的最小排列
do {
for(int i = 0; i < n; i++) printf("%d ", p[i]); // 输出排列p
printf("\n");
} while(next_permutation(p, p+n)); // 求下一个排列
return 0;
}
上面的代码同样可以用于可重集。
至此,枚举排列搞定了。我们主要可以学习到了递归搜索所有可能的情况,很暴力,但是很多情况下很有效。