LeetCode52——N-Queens II

最新推荐文章于 2025-01-02 19:21:53 发布

原创最新推荐文章于 2025-01-02 19:21:53 发布 · 1.2k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #N皇后 #回溯

leetcode 专栏收录该内容

195 篇文章

订阅专栏

这篇博客介绍了如何解决LeetCode第52题——N-Queens II，主要探讨了如何通过回溯算法来计算不同解决方案的数量。作者通过调整上一题（N-Queens）的代码，当行数等于n时，增加计数器而非添加解，从而实现解的数量统计。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

LeetCode52——N-Queens II

跟LeetCode51一样，只不过这题是求解解的个数。

这里我套用了上一题的代码，只不过当row==n（递归返回条件）时，不是将一种可行解加入结果集，而是另计数变量自增，来统计可行解的数量。

代码：

class Solution {
private:
	int num;
	bool check(int row, int col, int n, vector<string>m)
	{
		if (row == 0)
			return true;
		int i;
		int j;
		for (i = 0; i < row; i++)
		{
			if (m[i][col] == 'Q')
				return false;
		}
		i = row - 1;
		j = col - 1;
		while (i >= 0 && j >= 0)
		{
			if (m[i][j] == 'Q')
				return false;
			i--;
			j--;
		}
		i = row - 1;
		j = col + 1;
		while (i >= 0 && j < n)
		{
			if (m[i][j] == 'Q')
				return false;
			i--;
			j++;
		}
		return true;
	}
	void add(vector<string>m)
	{
		num++;
	}
	void solve(int row, int n, vector<string>m)
	{
		int col;
		if (row == n)
		{
			add(m);
			return;
		}
		for (col = 0; col<n; col++)
		{
			if (check(row, col, n, m) == true)
			{
				m[row][col] = 'Q';
				solve(row + 1, n, m);
				m[row][col] = '.';
			}
		}
	}
public:
		int totalNQueens (int n) {
		vector<string>m(n, string(n, '.'));
		num = 0;
		solve(0, n, m);
		return num;
	}
};