[LeetCode]N-Queens II

本题为非常典型的回溯问题，回溯问题的实质是从初始位置出发，层层推进问题，如果遇到到达某一位置时该位置不满足题设条件，则退回到上一层，在上一层中选择下一个位置继续进行试探。回溯问题相当于相对于枚举法进行了剪枝处理，一定程度上提高了效率。

这道题我完全没有做出来，全部是通过别人的答案进行拼凑，但中间还是出现了问题。代码如下：

public class Solution {
    public int count = 0;
	public int totalNQueens(int n) {
        int[] res=new int[n+1];
        for(int i=0;i<n;i++){
        	res[i]=-100;
        }
        countRow(0,res,n);
        return count;
    }
    private void countRow(int n,int[] res,int total){
        if(n==total)count++;
        for(int i=0;i<total;i++){
            boolean isqueen=true;
            for(int j=0;j<total;j++){
                if(!isQueen(n,i,j,res[j])){
                    isqueen=false;
                    break;
                }
            }
            if(isqueen){
                res[n]=i;
                countRow(n+1,res,total);
                res[n]=-100;
            }
        }
    }
    private boolean  isQueen(int cRow,int cCol,int pRow,int pCol){
        if(cCol==pCol||Math.abs(cRow-pRow)==Math.abs(cCol-pCol)){
            return false;
        }else{
            return true;
        }
    }
}

问题主要出现在：1、res数组要初始化，否则初始值为0会影响到isQueen的判断；2、进行递归时选择countRow(n+1,res,total)，而不是n++或者++n，因为之后递归完成返回后还要使用n进行下一位置的递归；3、isQueen里面cRow==pRow不是判断条件，因为没有相等行的存在。

上述代码还有点头重脚轻，再优化后如下：

public static int count = 0;
	public static int totalNQueens(int n) {
        int[] res=new int[n];
        for(int i=0;i<n;i++){
        	res[i]=-100;
        }
        countRow(0,res,n);
        return count;
    }
    private static void countRow(int n,int[] res,int total){
        if(n==total)count++;
        for(int i=0;i<total;i++){
            if(isQueen(n,i,res)){
                res[n]=i;
                countRow(n+1,res,total);
                res[n]=-100;
            }
        }
    }
    private static boolean  isQueen(int cRow,int cCol,int[] res){
    	for(int i=0;i<res.length;i++){
    		if(cCol==res[i]||Math.abs(cRow-i)==Math.abs(cCol-res[i])){
    			return false;
    		}
    	}
    	return true;
    }

回溯法之后还要反复练习。