本文实例讲述了C++的回溯法,分享给大家供大家参考之用。具体方法分析如下:
一般来说,回溯法是一种枚举状态空间中所有可能状态的系统方法,它是一个一般性的算法框架。
解向量a=(a1, a2, ..., an),其中每个元素ai取自一个有限序列集Si,这样的解向量可以表示一个排列,其中ai是排列中的第i个元素,也可以表示子集S,其中ai为真当且仅当全集中的第i个元素在S中;甚至可以表示游戏的行动序列或者图中的路径。
在回溯法的每一步,我们从一个给定的部分解a={a1, a2, ..., ak}开始,尝试在最后添加元素来扩展这个部分解,扩展之后,我们必须测试它是否为一个完整解,如果是的话,就输出这个解;如果不完整,我们必须检查这个部分解是否仍有可能扩展成完整解,如果有可能,递归下去;如果没可能,从a中删除新加入的最后一个元素,然后尝试该位置上的其他可能性。
用一个全局变量来控制回溯是否完成,这个变量设为finished,那么回溯框架如下,可谓是回溯大法之精髓与神器
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
bool finished = false ;
void backTack( int input[], int inputSize, int index, int states[], int stateSize)
{
int candidates[MAXCANDIDATE];
int ncandidates;
if (isSolution(input, inputSize, index) == true )
{
processSolution(input, inputSize, index);
}
else
{
constructCandidate(input, inputSize, index, candidates, &ncandidates);
for ( int i = 0; i < ncandidates; i++)
{
input[index] = candidates[i];
backTack(input, inputSize, index + 1);
if (finished)
return ;
}
}
}
|
不拘泥于框架的形式,我们可以编写出如下代码:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
#include <iostream>
using namespace std;
char str[] = "abc" ;
const int size = 3;
int constructCandidate( bool *flag, int size = 2)
{
flag[0] = true ;
flag[1] = false ;
return 2;
}
void printCombine( const char *str, bool *flag, int pos, int size)
{
if (str == NULL || flag == NULL || size <= 0)
return ;
if (pos == size)
{
cout << "{ " ;
for ( int i = 0; i < size; i++)
{
if (flag[i] == true )
cout << str[i] << " " ;
}
cout << "}" << endl;
}
else
{
bool candidates[2];
int number = constructCandidate(candidates);
for ( int j = 0; j < number; j++)
{
flag[pos] = candidates[j];
printCombine(str, flag, pos + 1, size);
}
}
}
void main()
{
bool *flag = new bool [size];
if (flag == NULL)
return ;
printCombine(str, flag, 0, size);
delete []flag;
}
|
采用回溯法框架来计算字典序排列:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
#include <iostream>
using namespace std;
char str[] = "abc" ;
const int size = 3;
void constructCandidate( char *input, int inputSize, int index, char *states, char *candidates, int *ncandidates)
{
if (input == NULL || inputSize <= 0 || index < 0 || candidates == NULL || ncandidates == NULL)
return ;
bool buff[256];
for ( int i = 0; i < 256; i++)
buff[i] = false ;
int count = 0;
for ( int i = 0; i < index; i++)
{
buff[states[i]] = true ;
}
for ( int i = 0; i < inputSize; i++)
{
if (buff[input[i]] == false )
candidates[count++] = input[i];
}
*ncandidates = count;
return ;
}
bool isSolution( int index, int inputSize)
{
if (index == inputSize)
return true ;
else
return false ;
}
void processSolution( char *input, int inputSize)
{
if (input == NULL || inputSize <= 0)
return ;
for ( int i = 0; i < inputSize; i++)
cout << input[i];
cout << endl;
}
void backTack( char *input, int inputSize, int index, char *states, int stateSize)
{
if (input == NULL || inputSize <= 0 || index < 0 || states == NULL || stateSize <= 0)
return ;
char candidates[100];
int ncandidates;
if (isSolution(index, inputSize) == true )
{
processSolution(states, inputSize);
return ;
}
else
{
constructCandidate(input, inputSize, index, states, candidates, &ncandidates);
for ( int i = 0; i < ncandidates; i++)
{
states[index] = candidates[i];
backTack(input, inputSize, index + 1, states, stateSize);
}
}
}
void main()
{
char *candidates = new char [size];
if (candidates == NULL)
return ;
backTack(str, size, 0, candidates, size);
delete []candidates;
}
|
对比上述两种情形,可以发现唯一的区别在于全排列对当前解向量没有要求,而字典序对当前解向量是有要求的,需要知道当前解的状态!
八皇后回溯法求解:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
#include <iostream>
using namespace std;
int position[8];
void constructCandidate( int *input, int inputSize, int index, int *states, int *candidates, int *ncandidates)
{
if (input == NULL || inputSize <= 0 || index < 0 || candidates == NULL || ncandidates == NULL)
return ;
*ncandidates = 0;
bool flag;
for ( int i = 0; i < inputSize; i++)
{
flag = true ;
for ( int j = 0; j < index; j++)
{
if ( abs (index - j) == abs (i - states[j]))
flag = false ;
if (i == states[j])
flag = false ;
}
if (flag == true )
{
candidates[*ncandidates] = i;
*ncandidates = *ncandidates + 1;
}
}
/*
cout << "ncandidates = " << *ncandidates << endl;
system("pause");*/
return ;
}
bool isSolution( int index, int inputSize)
{
if (index == inputSize)
return true ;
else
return false ;
}
void processSolution( int &count)
{
count++;
}
void backTack( int *input, int inputSize, int index, int *states, int stateSize, int &count)
{
if (input == NULL || inputSize <= 0 || index < 0 || states == NULL || stateSize <= 0)
return ;
int candidates[8];
int ncandidates;
if (isSolution(index, inputSize) == true )
{
processSolution(count);
}
else
{
constructCandidate(input, inputSize, index, states, candidates, &ncandidates);
for ( int i = 0; i < ncandidates; i++)
{
states[index] = candidates[i];
backTack(input, inputSize, index + 1, states, stateSize, count);
}
}
}
void main()
{
//初始化棋局
for ( int i = 0; i < 8; i++)
position[i] = i;
int states[8];
int count = 0;
backTack(position, 8, 0, states, 8, count);
cout << "count = " << count << endl;
}
|
希望本文所述对大家C++程序算法设计的学习有所帮助。