DFS解决八皇后问题

2019-07-29

16:49:15

#include <bits/stdc++.h>
using namespace std;
int mat[][];
int tot;

int check(int row, int col)
{
    for(int i = ; i <= row; i++)
    {
        if(mat[i][col] == ) //只需要判断在这一列是否已经方了
        {
            return ;
        }
        //判断主副对角线
        for(int j = ; j < ; j++)
        {
            if(mat[i][j] == )
            {
//                if (i + j == row + col )
//                {
//                    return 0;
//                }
//                else if(fabs(i - j) == fabs(row - col))
//                {
//                    return 0;
//                }
                if(fabs(i - row) - fabs(j - col) == ) //等腰梯形
                {
                    return ;
                }
                else
                {
//                    continue;
                    break;
                }

            }
        }

    }
    return ;
}

void dfs(int row)
{
    if(row == )
    {
        tot++;
        for(int i = ; i < ; i++)
        {
            for(int j = ; j < ; j++)
            {
                cout << mat[i][j] << " ";
            }
            cout << endl;
        }
        cout << endl;
    }
    else
    {
        for(int col = ; col < ; col++)
        {
            if(check(row, col) == )
            {
                mat[row][col] = ;
                dfs(row + );
                mat[row][col] = ;
            }
        }
    }
}
int main()
{
    dfs();
    cout << "tot = " << tot << endl;
    return ;

}