poj 1321 — 棋盘问题（简单DFS）

题目链接：http://poj.org/problem?id=1321

棋盘问题

Time Limit: 1000MS Memory Limit: 10000KTotal Submissions: 37101 Accepted: 18267

Description

在一个给定形状的棋盘（形状可能是不规则的）上面摆放棋子，棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列，请编程求解对于给定形状和大小的棋盘，摆放k个棋子的所有可行的摆放方案C。

Input

输入含有多组测试数据。 
每组数据的第一行是两个正整数，n k，用一个空格隔开，表示了将在一个n*n的矩阵内描述棋盘，以及摆放棋子的数目。 n <= 8 , k <= n 
当为-1 -1时表示输入结束。 
随后的n行描述了棋盘的形状：每行有n个字符，其中 # 表示棋盘区域， . 表示空白区域（数据保证不出现多余的空白行或者空白列）。 

Output

对于每一组数据，给出一行输出，输出摆放的方案数目C （数据保证C<2^31）。

Sample Input

2 1#..#4 4...#..#..#..#...-1 -1

Sample Output

21

Source

蔡错@pku

[Submit]   [Go Back]   [Status]   [Discuss]

Home Page   Go Back  To top

题目大意：略

解析:  #代表可以放棋子，用book[N} 标记下用过的列，扫描每一行，找到答案时，ans++

代码如下：

#include<iostream>#include<algorithm>#include<map>#include<stack>#include<queue>#include<vector>#include<set>#include<string>#include<cstdio>#include<cstring>#include<cctype>#include<cmath>#define N 19using namespace std;const int inf = 1e9;const int mod = 1<<30;const double eps = 1e-8;const double pi = acos(-1.0);typedef long long LL;int mp[N][N], ans, n, d, book[N];void dfs(int u, int cur){    if(cur == d) { ans++; return ; } // 达到要求，ans++， 返回    if(u == n + 1) return ;    for(int i = 1; i <= n; i++)    {        if(!book[i] && mp[u][i])        {            book[i] = 1;            dfs(u + 1, cur + 1);            book[i] = 0; // 回溯时用        }    }    dfs(u + 1, cur);//很重要，当前行没找到搜下一行，或者当前行搜完后搜下一行    return ;}int main(){    int i, j;    char s;    while(scanf("%d%d", &n, &d), n + d != -2)    {        memset(mp, 0, sizeof(mp));        for(i = 1; i <= n; i++)        {            for(j = 1; j <= n; j++)            {                scanf(" %c", &s);                if(s == '#') mp[i][j] = 1; // 1 表示能放棋子            }        }        ans = 0;        memset(book, 0, sizeof(book)); // 将标记数组初始化，1表示走过， 0表示没走过        dfs(1, 0);        cout << ans << endl;    }    return 0;}