八皇后问题

今天第一次写博客，就写一道dfs的基础问题（别骂了

八皇后问题描述如下

关于本道题，大致思路如下
设立dfs函数，参数x从1遍历到n，这里的x表示行数。然后在dfs内部进行列的遍历。

这是我做的第一道dfs，可以算是dfs启蒙了。从这一道题中，我初步认识到了深度优先搜索以递归为灵魂，比暴力枚举更加优化（因为剪掉了许多不必要的枝节）。

附上AC代码

#include<bits/stdc++.h>
using namespace std;

char cmap[15][15];//这里表示棋盘，有皇后则1，无则0
bool row_col[15],m_dia[30],c_dia[30];
//第一个是记录皇后所在的列能否放棋子，即某一列一旦有皇后则为0，第二个是主对角线即平行线能够放皇后，第三个是副对角线及其平行线，三者初始值均为1
int n,sum_case = 0;//sumcase是解的总数

void _set(int x,int y){//放置皇后
	cmap[x][y] = 1;
	row_col[y] = 0;//放一个皇后则三者均改为0
	m_dia[n - 1 +(x - y)] = 0;
	c_dia[x + y] = 0;
}

void take(int x,int y){//拿走皇后，与放皇后反
	cmap[x][y] = 0;
	row_col[y] = 1;
	m_dia[n - 1 +(x - y)] = 1;
	c_dia[x + y] = 1;
}

bool judge(int x,int y){//判断这个位置能不能放
	if(cmap[x][y] == 0 && row_col[y] == 1 && m_dia[n - 1 +(x - y)] == 1 && c_dia[x + y] == 1) return true;
	return false;
}

void output(){//输出
	if(sum_case >= 3) sum_case ++;
	else{
		for(int i = 0; i < n; i ++){
			for(int j = 0; j < n; j ++){
				if(cmap[i][j] == 1) cout<< j + 1 << ' ';
			}
		}
		cout << endl;
		sum_case ++;
	}
}
void dfs(int x){//x from 0 to n-1
	if(x == n){
		output();
	}
	else{
		for(int i = 0; i < n; i ++){
			if(judge(x,i)){
				_set(x,i);
				dfs(x + 1);
				take(x,i);
			}
		}
	}
}

int main(){
	cin >> n;
	memset(row_col,1,sizeof(row_col));
	memset(m_dia,1,sizeof(m_dia));
	memset(c_dia,1,sizeof(c_dia));
	memset(cmap,0,sizeof(cmap));
	dfs(0);
	cout << sum_case;
}