N皇后问题 LeetCode51. N-Queens

题意：在 n*n 的矩阵中摆放 n 个皇后，并且每个皇后不能在同一行，同一列，同一对角线上，求所有的 n 皇后的解。

题解：
一行一行地摆放，在确定一行中的那个皇后应该摆在哪一列时，需要用三个标记数组来确定某一列是否合法，这三个标记数组分别为：列标记数组、45 度对角线标记数组和 135 度对角线标记数组。

45 度对角线标记数组的长度为 2 * n - 1，通过下图可以明确 (r, c) 的位置所在的数组下标为 r + c。

135 度对角线标记数组的长度也是 2 * n - 1，(r, c) 的位置所在的数组下标为 n - 1 - (r - c)。

package BackTracking;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * 在 n*n 的矩阵中摆放 n 个皇后，并且每个皇后不能在同一行，同一列，同一对角线上，求所有的 n 皇后的解。
 * @author Archerlu
 *
 */
public class N_Queens {
	public static void main(String[] args) {
		int n = 8;
		List<int[][]> list = findN_Queens(n);
		for(int[][] arr:list) {
			for(int i=0;i<arr.length;i++) {
				System.out.println(Arrays.toString(arr[i]));
			}
			System.out.println("------------------------");
		}
	}

	private static List<int[][]> findN_Queens(int n) {
		List<int[][]> list = new ArrayList<>();
		if(n<3) return list;
		int[][] arr = new int[n][n];
		boolean[] colUsed = new boolean[n];
		boolean[] angle45 = new boolean[2*n-1];
		boolean[] angle135 = new boolean[2*n-1];
		backTracking(list,arr,colUsed,angle45,angle135,0);
		return list;
	}

	private static void backTracking(List<int[][]> list, int[][] arr, boolean[] colUsed, boolean[] angle45,
			boolean[] angle135, int row) {
		int n = arr.length;
		if(row==n) {
			list.add(copy(arr));
			return ;
		}
		for(int col=0;col<n;col++) {
			if(colUsed[col]||angle45[col+row]||angle135[n-1+row-col]) continue;
			arr[row][col] = 1;
			colUsed[col]=angle45[col+row]=angle135[n-1+row-col] = true;
			backTracking(list,arr,colUsed,angle45,angle135,row+1);
			arr[row][col] = 0;
			colUsed[col]=angle45[col+row]=angle135[n-1+row-col] = false;
		}
	}

	private static int[][] copy(int[][] arr) {
		int rows = arr.length;
		int cols = arr[0].length;
		int[][] arrCopy = new int[rows][cols];
		for(int i=0;i<rows;i++) {
			for(int j=0;j<cols;j++) {
				arrCopy[i][j] = arr[i][j];
			}
		}
		return arrCopy;
	}
}