LeetCode N-Queens II

最新推荐文章于 2022-06-06 10:35:33 发布

coobj2008

最新推荐文章于 2022-06-06 10:35:33 发布

阅读量309

点赞数

分类专栏：面试题

本文链接：https://blog.csdn.net/coobj2008/article/details/27231557

版权

面试题专栏收录该内容

144 篇文章 0 订阅

订阅专栏

题目：

Follow up for N-Queens problem.

Now, instead outputting board configurations, return the total number of distinct solutions.

class Solution {
public:
	int totalNQueens(int n) {
		res = 0;
		memset(used, false, sizeof(used));
		dfs(0, n);
		return res;
	}
private:
	int res;
	// a[i]表示第i行的a[i]列有Queen
	int a[100];
	int used[100];
	void dfs(int dep, int maxDep) {
		if (dep == maxDep) {
			res++;
			return;
		}
		for (int y = 0; y < maxDep; y++) {
			if (check(dep, y) && !used[y]) {
				used[y] = true;
				a[dep] = y;
				dfs(dep + 1, maxDep);
				used[y] = false;
			}
		}
	}
	bool check(int x, int y) {
		for (int i = 0; i < x; i++) {
			if (abs(i-x) == abs(y-a[i]))
				return false;
		}
		return true;
	}
};