051N-Queens

写到subsets，顺便回想了一下8皇后问题，写个答案记录一下吧

每次从列开始遍历，1皇后放在第一行第一列，2皇后放在第二行第一列，是否ok？第二列是否ok？ok后进入下一行，3皇后放在第三行第一列是否ok？第二列？。。。全部放完之后回到1皇后第一行第二列.

判断条件有点可以优化时间度的，懒得搞了先这样吧。。

class Solution {
    public static List<List<String>> ans = new ArrayList<>();

	// 数组防止皇后位置，第n行的皇后为皇后n，放在第array[n]列
	public static int[] array = new int[100];

    public List<List<String>> solveNQueens(int n) {
        ans.clear();
		dfs(0, n);
		return ans;

    }
    
    	public static void dfs(int index, int n) {
		if (index >= n) {
			List<String> chess = new ArrayList<>();
			for (int i = 0; i < n; i++) {
				String str = "";
				for (int j = 0; j < n; j++) {
					if (j == array[i]) {
						str += "Q";
					} else {
						str += ".";
					}
				}
				chess.add(str);
			}
			ans.add(chess);
		}
		// 遍历每一列，如果ok，检查下一列
		for (int i = 0; i < n; i++) {
			// 第index行的皇后每次从第一列开始检测是否ok
			array[index] = i;
			if (judge(index)) {
				dfs(index + 1, n);
			}
		}

	}

	private static boolean judge(int n) {  
        for (int i = 0; i < n; i++) {  
            if (array[i] == array[n] || Math.abs(n - i) == Math.abs(array[n] - array[i])) {  
                return false;  
            }  
        }  
        return true;  
    }  

}