回溯算法的一些问题
程序员文章站
2022-05-20 22:30:51
...
八皇后问题
在对角线上这个条件可以抽象成数学的直线,就是倒数等于1的一条直线。所以会有 abs(y1 - y2) = abs(x1 - x2)
void Search(int cur)
{
if(cur == 8){tot++;}//满足条件就让让答案加一
else for(int i = 0;i < 8;i++){
int ok = 1;
c[cur] = i;
for(int j = 0;j < cur;j++){//检查是否和前面的皇后冲突
if(c[cur] == c[j] || abs(c[j] - c[cur]) == cur - j){//在对角线上倒数等于1或者等于-1
ok = 0;break;
}
}
if(ok)Search(cur + 1);//如果不冲突则继续递归
}
}
如果你发现了每一条主对角线的横纵坐标之差等于一个常数,即(x1,y1),(x2,y2)如果在一条主对角线上就会有y1 - x1 = y2 - x2;并且副对角线之和等于一个常数,就可以使用标记数组进行优化了
用标记数组进行优化
void Search2(int cur)
{
if(cur == 8){tot2++;}
else for(int i = 0;i < 8;i++){
if(!vis[0][i] && !vis[1][cur + i] && !vis[2][cur - i + 35]){
vis[0][i] = vis[1][cur + i] = vis[2][cur - i + 35] = 1;
Search2(cur + 1);
vis[0][i] = vis[1][cur + i] = vis[2][cur - i + 35] = 0;
}
}
}