欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

回溯算法的一些问题

程序员文章站 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;
        }
    }
}
相关标签: 回溯