欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

N皇后问题---C++实现

程序员文章站 2024-02-24 11:32:40
...

利用dfs和回溯思想,关键是要利用好3个条件:
1.所在列没有皇后
2.所在位置主对角线上没有元素
3.所在位置斜对角线上没有元素 利用状态数组保存这3个条件的信息

int num = 0;
void dfs(vector<vector<int> >& res,vector<vector<int> >&flag ,int level) {
	if (level == res.size()) {
		for (unsigned int i = 0; i < res.size(); i++) {
			for (unsigned int j = 0; j < res[0].size(); j++) {
				cout << res[i][j] << " ";
			}
			cout << endl;
		}
		num++;
		cout << endl << endl;
	}
	else {
		for (unsigned int i = 0; i < res[0].size(); i++) {

//!flag[0][i] 表示第i列没有皇后
//!flag[1][level + i] 表示该位置的主对角线上没有皇后
//!flag[2][level - i + res[0].size()] 表示该位置的斜对角线上没有皇后

			if (!flag[0][i] && !flag[1][level + i] && !flag[2][level - i + res[0].size()]) {
				res[level][i] = 1;
				flag[1][level + i] = 1;
				flag[0][i] = 1;
				flag[2][level - i + res[0].size()] = 1;
				dfs(res, flag, level + 1);
				res[level][i] = 0;
				flag[1][level + i] = 0;
				flag[0][i] = 0;
				flag[2][level - i + res[0].size()] = 0;
			}
		}
	}

}

相关标签: 回溯

上一篇: Linux文件内容去重

下一篇: