题目:
棋盘问题
Time Limit: 1000MS | Memory Limit: 10000K | |
Total Submissions: 53363 | Accepted: 25784 |
Description
在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,请编程求解对于给定形状和大小的棋盘,摆放k个棋子的所有可行的摆放方案C。
Input
输入含有多组测试数据。
每组数据的第一行是两个正整数,n k,用一个空格隔开,表示了将在一个n*n的矩阵内描述棋盘,以及摆放棋子的数目。 n <= 8 , k <= n
当为-1 -1时表示输入结束。
随后的n行描述了棋盘的形状:每行有n个字符,其中 # 表示棋盘区域, . 表示空白区域(数据保证不出现多余的空白行或者空白列)。
每组数据的第一行是两个正整数,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
2 1
Source
蔡错@pku
思路:数据给的不大,直接DFS暴力搜索。
由于任意的两个棋子不能摆放在同一行或者同一列,那么可用一个一维数组vis记录第i行/第i列是否被使用过。并且只需要从第0行搜索到第n-1行。对于每一行的可摆放位置,若要在此摆放棋子,那么vis数组对应位置要标记使用过。但也有可能不在这个位置摆放棋子,则在以当前位置摆放棋子的前提下搜索完成后,vis数组的当前位置状态要回溯至未摆放的状态继续搜索。
dfs结束标志为搜索完第n-1行。
我的日常踩坑:1。cin和scanf不能混用。否则就是无限WA
2。po这种古老的OJ不支持#include<bits/stdc++.h>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<map>
#include<vector>
#include<stack>typedef long long ll;using namespace std;int n,k;char mapp[10][10];bool vis[10]; //vis数组记录第i个位置是否被使用过 int ans;void dfs(int line, int sum){if(line == n){if(sum == k){ //若0-n行的摆放棋子总数==k,则总摆放数++ ans++;}return;}for(int i = 0 ; i<n ; i++){if(vis[i] == 0 && mapp[line][i]=='#'){vis[i] = 1;dfs(line+1 , sum+1); //这两行表示:若i位置可用 //则在第line行第i个位置摆放棋子,继续dfs vis[i] = 0; //状态回溯 }}dfs(line+1 , sum); //不在第line行的任何一个位置摆放棋子
} int main(){ios::sync_with_stdio(false);while (scanf("%d %d",&n,&k)!=EOF){if(n==-1 && k == -1) break;for(int i = 0 ; i<n ; i++){scanf("%s",mapp[i]);}ans = 0;memset(vis,0,sizeof(vis));dfs(0,0);printf("%d\n",ans);}return 0;
}