POJ-1321 棋盘问题
题目链接:POJ-1321
题目大意:给定一个棋盘 棋盘#的位置可以放置棋子 要求你放入k个棋子 并且放过棋子的同一行同一列不能再放棋子 问有多少种放法
解题思路:将可以设置棋子的地方放入一个容器中 之后递归往下看能不能放即可,当然还要试试当前要放的位置横行和列数有没有放过 当放k个棋子的时候总数加一 最后输出即可 这题时间要求挺苛刻的吧 最大行列数的时候我的递归基本都要比网上的代码少调用三四千次,但是就是超时,最后加了一句当判断到最后一行的点时直接返回才通过
代码块:
#include<iostream>
#include<set>
#include<vector>
using namespace std;
int n,k;
char arr[10][10] = {'\0'};
int sum = 0;
typedef pair<int, int> p;
set<int> setA;
vector<p> vectorA;
void dfs(int num, int x);
int main() {
while(cin>>n>>k) {
if(n == -1 && k == -1) break;
for(int i = 1; i <= n; i++) {
for(int j = 1; j <= n; j++) {
cin>>arr[i][j];
if(arr[i][j] == '#') {
vectorA.push_back(p(i,j));
}
}
}
for(int i = 0; i < vectorA.size(); i++) {
if(vectorA.size() >= i+k) {
dfs(k-1,i);
}
}
cout<<sum<<endl;
sum = 0;
vectorA.clear();
setA.clear();
}
return 0;
}
void dfs(int num, int x) {
if(num == 0) {
sum++;
return;
}
if(vectorA[x].first == n) {
return;
}
setA.insert(vectorA[x].second);
for(int i = x + 1; i < vectorA.size(); i++) {
if(vectorA[i].first == vectorA[x].first) continue;
if(setA.find(vectorA[i].second) == setA.end()) {
dfs(num-1, i);
}
}
setA.erase(vectorA[x].second);
}