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

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

int num = 0;
void dfs(vector >& res,vector >&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;
			}
		}
	}

}

你可能感兴趣的:(算法题,数据结构,c/c++面试,回溯)