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文件内容去重