N皇后问题 – 解法二（斜线判断改进）

import java.util.*;
import java.math.*;

public class Solution {

    HashSet<Integer> res = new HashSet<Integer>();
    //已经摆放几个
    int leng = 0;
    //多少种摆法
    int count = 0;
    int len = 0;
    TreeMap<Integer, Integer> chess = new TreeMap<Integer, Integer>();
    int midNum = 0;
    /**
     *
     * @param n int整型 the n
     * @return int整型
     */
    public int Nqueen (int n) {
        len = n;
        recursion();
        return res.size();
    }
    private void recursion() {

        if (chess.size() == len) {
            midNum = 0;
            //元素顺序不同，迭代的直不同，所以要去重的话，只能使用treemap排序
            for (Map.Entry<Integer, Integer> en : chess.entrySet()) {
                midNum = (31 * midNum + en.getKey()) + en.getValue();
            }
            res.add(midNum);
            return;
        }
        
        for (int x = 1 ; x <= len; x++ ) {
            for (int y = 1 ; y <= len ; y++) {
                if (!find(x, y)) {
                    chess.put(x, y);
                    //进入下一个寻找Q的迭代
                    recursion();
                    //每确定一个Q的位置，回溯
                    //回溯只能保证当前Q进入其他分支，但是其他Q仍然可以在当前分支
                    //而每个Q的位置要考虑的是组合而不是排列
                    chess.remove(x);
                    leng--;
                    //一行只能有一个Q
                    break;
                }
            }
        }


    }
    private boolean find(int x, int y) {

        for (Map.Entry<Integer, Integer> en : chess.entrySet()) {
            if (en.getKey() == x ||
                    en.getValue() == y ||
                    Math.abs(x - en.getKey()) == Math.abs(y - en.getValue())
               ) {
                return true;
            }
        }
        return false;


    }
}