利用递归实现不定重数多重循环(附源代码)
很多情况下我们要实现的程序本身并不复杂但却很烦琐,这里举一个穷举的例子。多数穷举程序需要遍历多个循环点,我们遇到的情况经常是:变量a的变化范围是aMin~aMax,在a的每个取值上b又要从bMin~bMax全都遍历一遍……如果是只有a,b两个变量那实现起来当然方便,只需如下代码即可:
for(a = aMin; a < aMax; a += da)
{
for(b = bMin; b < bMax; b += db)
{
// do some thing
...
}
}
但是如果循环点比较多,甚至是一个数组呢?你会说这好办,只要:
for(a = aMin; a < aMax; a += da)
{
for(b = bMin; b < bMax; b += db)
{
for(c = cMin; c < cMax; c += dc)
{
for(d = dMin; d < bMax; d += dd)
{
for(e = eMin; e < eMax; e += de)
{
for(f = fMin; f < fMax; f += df)
{
...
}
}
}
}
}
}
……也许可行吧……
不过,这篇文章要向大家推荐一种更为方便的方法——递归。 递归,说白了就是自己调用自己的函数。利用递归这种特殊的机制,我们可以把上面大量重复的代码化繁为简。具体实现方法可以参考以下模型:
bool Recursion(int index, int min, int max)
{
for(int i = min; i <= max; i++)
{
if(index == 0)
{
// 在这里添加搜索代码,如果要中途终止搜索,则返回true
...
return true or false;
}
else
{
if(Recursion(index - 1, min, max))
{
// 收到搜索终止信号,直接跳出
return true;
}
}
}
return true;
}
这里index表示调用的层数,可以用来在一开始调用这个函数时控制多重循环的重数,min和max是每重循环的搜索范围,可以根据实际需要写出每重循环的搜索范围都不同的多重循环。
当然这个只是最一般的模型,可以根据实际问题的需要进行变化。例如,如果min和max不会变化,可以在程序中写死;index也可以从0开始到特定值时结束。
下面举一个实际的例子:
以下这段代码实现了在城市数目较少的情况下,利用穷举法解决旅行商问题(tsp)(城市数目较多的情况可以参考我的另一篇文章《用遗传算法解决旅行商问题(附源代码)》)。这个程序在我的笔记本上运行11个或以下城市的tsp用时小于1秒,12个城市的tsp需要5~7秒,还是可以接受的。城市再多,已经没有什么实际意义,顶多还可以作为教学试验。不过城市数量较少的情况下穷举法也是一种非常具有竞争力的算法,至少它求出来的是确切全局最优解。
#include <stdio.h>
#include <math.h>
#include <memory.h>
#include <time.h>
#include <conio.h>
#define MAX_NUM 20
double s_citys[MAX_NUM][2];
double s_distances[MAX_NUM][MAX_NUM];
int s_num = 0;
double s_min = 1e200;
int s_indexMin[MAX_NUM];
int s_index[MAX_NUM];
bool s_mark[MAX_NUM];
bool ReadFile()
{
FILE * fp = fopen("citys.txt", "r");
if(fp)
{
s_num = 0;
while(!feof(fp) && s_num < MAX_NUM)
{
fscanf(fp, "%lf, %lf/n", &s_citys[s_num][0], &s_citys[s_num][1]);
s_num++;
}
fclose(fp);
return true;
}
return false;
}
void Search(int index)
{
int i, j;
double sum;
for(i = 1; i < s_num; i++)
{
if(s_mark[i])
{
s_index[index] = i;
s_mark[i] = false;
if(index == s_num - 1)
{
sum = 0.0;
for(j = 0;j < s_num - 1; j++)
{
sum += s_distances[s_index[j]][s_index[j + 1]];
}
sum += s_distances[s_index[s_num - 1]][s_index[0]];
if(sum < s_min)
{
memcpy(s_indexMin, s_index, sizeof(s_index));
s_min = sum;
}
}
else
{
Search(index + 1);
}
s_mark[i] = true;
}
}
}
void WriteFile()
{
FILE * fp = fopen("result.txt", "w");
if(fp)
{
for(int i = 0; i < s_num; i++)
{
fprintf(fp, "%d/n", s_indexMin[i]);
}
fclose(fp);
}
}
void main()
{
int i, j;
if(ReadFile())
{
time_t start, end;
double timeUsed;
time(&start);
printf("开始穷举,请等待...");
for(i = 1; i < s_num; i++)
{
for(j = 0; j < i; j++)
{
s_distances[i][j] =
sqrt(
(s_citys[i][0] - s_citys[j][0]) * (s_citys[i][0] - s_citys[j][0]) +
(s_citys[i][1] - s_citys[j][1]) * (s_citys[i][1] - s_citys[j][1]));
s_distances[j][i] = s_distances[i][j];
}
}
s_min = 1e200;
s_index[0] = 0;
memset(s_mark, true, sizeof(s_mark));
s_mark[0] = false;
Search(1);
time(&end);
timeUsed = difftime(end, start);
if(timeUsed < 1.0)
{
printf("/n用时小于1秒!/n");
}
else
{
printf("/n用时%.0lf秒!/n", timeUsed);
}
printf("/n最短行程%lf/n", s_min);
printf("最佳访问序列:/n");
for(int i = 0; i < s_num; i++)
{
printf("%02d/n", s_indexMin[i]);
}
WriteFile();
printf("按任意键继续...");
getch();
}
else
{
printf("读取文件失败");
}
}
程序的输入是一个名为citys.txt的文本文件,里面记录了城市的坐标(一般默认坐标的范围为(0.0,0.0)~(1.0,1.0))。输出是一个名为result.txt的文本文件,里面记录了最佳旅行路线所经过的城市序号顺序(城市序号为0~n-1,n为城市数目)。下面是这两个文件的一个样例:
citys.txt
0.930000, 0.620000
0.590000, 0.870000
0.430000, 0.440000
0.560000, 0.480000
0.050000, 0.300000
0.740000, 0.860000
0.130000, 0.210000
0.070000, 0.380000
0.690000, 0.120000
0.970000, 0.430000
0.850000, 0.730000
0.800000, 0.360000
result.txt
0
9
11
8
6
4
7
2
3
1
5
10