今天学习了回溯法,其基本思想在于构建解空间树,并尽可能的剪枝。
一般设解向量为X(x1,x2,x3.......xn)求子集。
四色问题,亮点在于把关联的图形转化为点线相连的图论,再回溯。
例题有挺多:
1. 背包问题的回溯解法,即用部分背包来做剪枝优化。
2. 放缩上下界,剪枝。
3. 备忘录方法
4. 在已排列过的数字不再选用,使用全排列或者哈希。
下面就摆出N皇后问题的一种解:
#include <stdio.h>
int sum;
int total;
int arr[100];
int pp[11];
void swap(int *a,int *b)
{
int temp;
temp=*a;
*a=*b;
*b=temp;
}
void perm(int n)
{
if(n>total)
{
sum++;
return;
}
int i,j,flag;
for(i=n;i<=total;i++)
{
swap(arr+n,arr+i);
flag=0;
for(j=1;j<n;j++) //内层循环用于判断不再一条斜线上,即斜率不等于1,-1.全排列已经满足了不再同行同列的要求
{
if(n-j == arr[n]-arr[j])
{flag=1;break;}
if(n-j == arr[j]-arr[n])
{flag=1;break;}
}
if(flag==1)
{
swap(arr+n,arr+i);
continue;
}
perm(n+1);
swap(arr+n,arr+i);
}
}
int main()
{
int n,i,j;
for(i=1;i<=10;i++)
arr[i]=i;
for(i=1;i<=10;i++)
sum=0,total=i,perm(1),pp[i]=sum;
while(scanf("%d",&n) && n!=0)
{
printf("%d\n",pp[n]);
}
return 0;
}
上面的方法是通过全排列得出的解,还可以通过设置矩阵标志周边位置的方法
#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;
int vis[3][50], P[15];//三个方向 ↖↑↗在此三个方向都不能有皇后
int n, sum;
void DFS(int row);
int main()
{
for(n = 1; n <= 10; n++)//先打表不然会超时的
{
memset(vis,0,sizeof(vis));
sum = 0;
DFS(1);
P[n] = sum;
}
while(scanf("%d",&n), n)
{
printf("%d\n",P[n]);
}
return 0;
}
void DFS(int row)
{
int i;
if(row == n + 1)//已经够n行了
{
sum ++;
return ;
}
for(i = 1; i <= n; i++)
{
if(vis[0][row-i+n] == 0 && vis[1][i] == 0 && vis[2][row+i] == 0)
{//学习
vis[0][row-i+n] = vis[1][i] = vis[2][row+i] = 1;//变值
DFS(row + 1);//深搜
vis[0][row-i+n] = vis[1][i] = vis[2][row+i] = 0;//回溯
}
}
}