poj1321 棋盘问题（DFS）

题目链接

http://poj.org/problem?id=1321

题意

给定一块棋盘（棋盘可能是不规则的），有k个相同棋子，将k个棋子摆放在棋盘上，使得任意两个棋子不同行，不同列，求有多少种不同的摆法。

思路

此题和八皇后问题很像，需要注意摆放完毕后要恢复现场，便于下一次摆放。

代码

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int N = 10;
char board[N][N];
int visit[N];   //visit[j]=1表示第j列有棋子， 0表示无棋子
int n, k;
int cnt;       //摆放方案数

void dfs(int k, int r)
{
    if(k==0)
    {
        cnt++;
        return;
    }

    for(int i=r; i<N; i++)
    {
        for(int j=0; j<N; j++)
        {
            if(board[i][j]=='#'&& !visit[j])
            {
                visit[j] = 1;
                dfs(k-1, i+1);  //从下一行开始搜索
                visit[j] = 0;   //恢复现场，便于回溯
            }
        }
    }
}

int main()
{
    //freopen("poj1321.txt", "r", stdin);
    while(cin>>n>>k)
    {
        if(n==-1)
            return 0;
        for(int i=0; i<n; i++)
            cin>>board[i];

        memset(visit, 0, sizeof(visit));
        cnt = 0;
        dfs(k, 0);
        cout<<cnt<<endl;
    }
    return 0;
}