LeetCode：51.N皇后

典型的回溯法思路：从第一行开始，取第一个列，判断是否可以填Q，可以的话，进入第二层，依次选列，如果可以填Q，则下探到下一层，如果不可以，则回溯。

关键：如何判断是否可以填Q，即当前(row, col)是否合法，可以从数学坐标斜率上发现如下规律：

row+col = 常数 副对角线的标识
col-row = 常数 主对角线的标识

用三个set集合存储前面填入了Q之后，那些列，主副对角线的特征值

注意回溯之后，要对相应的set集合移出下一层填入的值

@Test
public void test() {
	List<List<String>> ans = solveNQueens(8);
	
	for(List<String> l : ans) {
		for(int i=0; i<l.size(); i++) {
			if(i==l.size()-1) {
				System.out.println("----------------------------");
			}
			System.out.println();
			System.out.println(l.get(i));
		}
	}
}

private Set<Integer> cols = new HashSet<>();
private Set<Integer> main = new HashSet<>();
private Set<Integer> assist = new HashSet<>();
private List<List<String>> result = new ArrayList<>();

public List<List<String>> solveNQueens(int n) {
	solve(n, 0, new ArrayList<String>());
	return result;
}

private void solve(int n, int row, List<String> list) {
	if(row==n-1) {
		for(int col=0; col<n; col++) {
			if(isValid(row, col)) {
				list.add(printRow(col, n));
				result.add(list);
			}
		}
		return;
	}
	
	for(int col=0; col<n; col++) {
        //合法往下一层走，不合法往下一列走
		if(!isValid(row, col)) {
			continue;
		}
		
		//下探一层
		List<String> tmp = new ArrayList<>();
		tmp.addAll(list);
		tmp.add(printRow(col, n));
		cols.add(col);
		main.add(col-row);
		assist.add(row+col);
		
		solve(n, row+1, tmp);
		
		//回溯，状态回滚
		cols.remove(col);
		main.remove(col-row);
		assist.remove(row+col);
	}
	
}

private String printRow(int col, int n) {
	//col=0;n=4->. . . Q
	//col=2;n=4->. . Q .
	StringBuilder sb = new StringBuilder();
	for(int i=0; i<n; i++) {
		if(i==col) {
			sb.append("Q");
		} else {
			sb.append(".");
		}
	}
	return sb.toString();
}

//判断当前row, col位置是否合法
private boolean isValid(int row, int col) {
	if(cols.contains(col) || main.contains(col-row) || assist.contains(row+col)) {
		return false;
	}
	return true;
}