力扣51.N皇后

题目:biubiu
题意:在nXn的棋盘中放n个皇后,要求n个皇后不能相互攻击,在提示中指出,两个皇后不能相同行相同列,并且不能再同一条斜线上,这个题不难,但是要认真,这个题只需要用深搜,进行递归回溯就行,但是在回溯的时候一定要注意一个小细节,你是如何标记不能再同一条斜线上的,我用的一个二维数组,将这个皇后的下方斜线全都标记,但是这就有一种可能,两条斜线会有相交的点,如果你直接标记为1,回溯直接回溯成0,那就有可能会造成一点的其他斜线的忽略,所以标记的时候不要直接标记为1,可以进行++,这样某一个点回溯的时候对其他点不会造成影响。

#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<algorithm>
using namespace std;
class Solution {
    
    
public:
	/*void print(vector<vector<int>>vis,int x) {
		cout << "**************************************************" << endl;
		cout << x << endl;
		for (int i = 0; i < vis.size(); i++) {
			for (int j = 0; j < vis[i].size(); j++)
				cout << vis[i][j] << " ";
			cout << endl;
		}
		cout << "**************************************************" << endl;
	}*/
	void dfs(vector<vector<string>>&ans, vector<string>p, vector<int>&lie, vector<vector<int>>&vis,int n,int x) {
    
    
		//print(vis,x);
		if (x == n) {
    
    
			for (int i = 0; i < n; i++) {
    
    
				if (lie[i] == 0&&vis[x-1][i]==0) {
    
    
					p[x - 1][i] = 'Q';
					ans.push_back(p);
					p[x - 1][i] = '.';
				}
			}
			return;
		}
		for (int i = 0; i < n; i++) {
    
    
			if (lie[i] == 0&&vis[x-1][i]==0) {
    
    
				lie[i] = 1;
				p[x - 1][i] = 'Q';
				int j = i , k = x-1;
				while (j < n && k < n ) {
    
    
                          vis[k][j] ++;
						  k++;
						  j++;
				}
				k = x-1;
				j = i ;
				while (k < n && j >= 0) {
    
    
					vis[k][j] ++;
					k++;
					j--;
				}
				dfs(ans, p, lie,vis, n, x + 1);
				j = i ;
				k = x-1;
				while (j < n && k < n) {
    
    
					vis[k][j] --;
					k++;
					j++;
				}
				k = x-1;
				j = i ;
				while (k < n && j >= 0) {
    
    
					vis[k][j]--;
					k++;
					j--;
				}
				lie[i] = 0;
				p[x - 1][i] = '.';
			}
		}
		return;
	}
	vector<vector<string>> solveNQueens(int n) {
    
    
		vector<vector<string>> ans;
		vector<string>p;
		for (int i = 0; i < n; i++) {
    
    
			string p1;
			p1.append(n, '.');
			p.push_back(p1);
		}
		vector<int>lie(n, 0);
		vector<vector<int>>vis(n, vector<int>(n, 0));
		dfs(ans, p, lie,vis, n, 1);
		return ans;
	}
};

int main() {
    
    
	int n;
	while (cin >> n) {
    
    
		Solution s;
		vector<vector<string>>ans;
		ans=s.solveNQueens(n);
		for (int i = 0; i < ans.size(); i++) {
    
    
			for (int j = 0; j < n; j++) {
    
    
				for(int k=0;k<n;k++)
				    cout << ans[i][j][k] << " ";
				cout << endl;
			}
			cout << endl;
		}
	}
	return 0;
}

おすすめ

転載: blog.csdn.net/qq_43840681/article/details/121275750