[leetcode]N-Queens II

一开始不明白为什么有这么一道几乎完全一样的，但把I的改一下超时了，所以要优化一下。经搜索，发现可以这样优化：使用col[i]表示第i行Q所在的列，那么，判断斜线是否满足就可以用abs(k-i)== abs(col[k]-j)，注意这里k要小于i，因为后面的行没有放Q，而默认初始化时0列放Q，就会错误。

public class Solution {

    private int ans;

    private boolean[] visited;

    private int[] col;

    public int totalNQueens(int n) {

        ans = 0;

        col = new int[n];

        visited = new boolean[n];

        sub(0);

        return ans;

    }

    

    private void sub(int i)

    {

        int n = col.length;

        if (i == n)

        {

            ans++;

        }

        else

        {

            for (int j = 0; j < n; j++)

            { 

                if (!visited[j] && isValid(i, j))

                {

                    col[i] = j;

                    visited[j] = true;

                    sub(i+1);

                    visited[j] = false;

                } 

            }

        }

    }

    

    private boolean isValid(int i, int j)

    {

        // only need to validate column and 'X' direction

        for (int k = 0; k < i; k++)

        {

            if (Math.abs(k-i)== Math.abs(col[k]-j)) return false;

        }

        return true;

    }

}