【算法入门经典】7.4回溯法【八皇后问题】

7.4.1

方法1.

#include<stdio.h>
#include<string.h>
int ans,n, map[20];
void search(int cur)
{
    int i,j,flag;
    if(cur == n+1)//递归边界 
        ans ++;
    else
    {
        for(i = 1; i <= n; i ++)
        {
            map[cur] = i;//把第cur行的皇后放在第i列 
            flag = 1;
            for(j = 1; j < cur; j ++)
            {//判断是否和前面的皇后冲突 
                if(map[cur] == map[j]||cur-map[cur]==j-map[j]||cur+map[cur]==j+map[j])
                {
                    flag = 0;break;
                }
            }
            if(flag)//如果合法，则继续递归 
                search(cur+1); 
        }
    }
}
int main()
{
    while(scanf("%d",&n),n!=0)
    {
        ans = 0;
        memset(map,0,sizeof(map));
        search(1);
        printf("%d\n",ans);
    }
    return 0;
}

方法2

#include<stdio.h>
#include<string.h>
#define N 100
int ans, n ,vis[3][100];
void dfs(int cur)
{
    int i;
    if( cur == n+1)
        ans ++;
    else
    {
        for(i = 1; i <= n; i ++)
        {
            if(!vis[0][i]&&!vis[1][cur+i]&&!vis[2][cur-i+n])//利用二维数组直接判断 
            {//该列没有放过棋子&&主对角线没有放过棋子&&副对角线没有放过棋子 
                vis[0][i] = vis[1][cur+i] = vis[2][cur-i+n] = 1;//修改全局变量 
                dfs(cur+1);
                vis[0][i] = vis[1][cur+i] = vis[2][cur-i+n] = 0;//回溯 
            }
        }
    }
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        ans = 0;
        memset(vis,0,sizeof(vis));
        dfs(1);
        printf("%d\n",ans);
    }
    return 0;
}