八皇后问题（对角线的处理）

1007.8皇后问题

时限：1000ms 内存限制：10000K  总时限：3000ms

描述

输出8皇后问题所有结果。

输入

没有输入。

输出

每个结果第一行是No n：的形式，n表示输出的是第几个结果；下面8行，每行8个字符，‘A’表示皇后，‘.’表示空格。不同的结果中，先输出第一个皇后位置靠前的结果；第一个皇后位置相同，先输出第二个皇后位置靠前的结果；依次类推。

输入样例

输出样例

输出的前几行：
No 1:
A.......
....A...
.......A
.....A..
..A.....
......A.
.A......
...A....
No 2:
A.......
.....A..
.......A
..A.....
......A.
...A....
.A......
....A...

提示

输出样例是正确输出结果的前几行。

来源

#include <iostream>
#include <cstdio>
#include <cstring>
#include <map>
using namespace std;
int checkrow[10],checkcol[10];
map<int,int> si;
char str[10][10];
int cas;
void dfs(int cnt)
{
    int i,j;
    if(cnt==9)
    {
        printf("No %d:\n",cas++);
        for(i=1;i<=8;i++)
        {
             for(j=1;j<=8;j++)
             printf("%c",str[i][j]);
             printf("\n");
        }
        return;
    }
    for(i=1;i<=8;i++)
    {
        if(checkrow[cnt]!=0||checkcol[i]!=0||si[i-cnt]!=0||si[10*(i+cnt)]!=0) continue; //对角线的处理 主对角线是列减行，从对角线是10*(行+列）来记录，避免重复
            str[cnt][i]='A';
            checkrow[cnt]=1;
            checkcol[i]=1;
            si[i-cnt]=1;
            si[10*(i+cnt)]=1;
            dfs(cnt+1);
            checkrow[cnt]=0;
            checkcol[i]=0;
            si[i-cnt]=0;
            si[10*(i+cnt)]=0;
            str[cnt][i]='.';
    }
}
int main()
{
   int i,j;
   for(i=1;i<=8;i++)
   for(j=1;j<=8;j++)
   str[i][j]='.';
   memset(checkrow,0,sizeof(checkrow));
   memset(checkcol,0,sizeof(checkcol));
   cas=1;
   dfs(1);
   return 0;
}