POJ-1321 棋盘问题

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);
}