关闭

[刷题]N-Queens

169人阅读 评论(0) 收藏 举报
分类:

[LintCode]N-Queens

class Solution {
    /**
     * Get all distinct N-Queen solutions
     * @param n: The number of queens
     * @return: All distinct solutions
     * For example, A string '...Q' shows a queen on forth position
     */
    ArrayList<ArrayList<String>> solveNQueens(int n) {
        // 2015-07-07
        ArrayList<ArrayList<String>> rst = new ArrayList<>();
        if (n == 0) {
            return rst;
        }
        ArrayList<Integer> list = new ArrayList<>();
        helper(rst, list, n);
        return rst;
    }
    
    private void helper(ArrayList<ArrayList<String>> rst, ArrayList<Integer> list ,int n) {
        // 控制树的深度
        if (list.size() == n) {
            rst.add(convert(list));
            return;
        }
        
        for (int i = 0; i < n; i++) {
            // 控制分支
            if (!isValid(list, i)) {
                continue;
            }
            list.add(i);
            helper(rst, list, n);
            list.remove(list.size() - 1);
        }
        return;
    }
    
    // 将数组转为string组
    private ArrayList<String> convert(ArrayList<Integer> list) {
        ArrayList<String> rst = new ArrayList<>();
        for (int i = 0; i < list.size(); i++) {
            String s = new String("");
            for (int j = 0; j < list.size(); j++) {
                if (j == list.get(i)) {
                    s += "Q";
                } else {
                    s += ".";
                }
            }
            rst.add(s);
        }
        return rst;
    }
    
    // 判断是queen是否可以相互攻击
    private boolean isValid(ArrayList<Integer> list, int k) {
        if (list.size() == 0) {
            return true;
        }
        for (int i = 0; i < list.size(); i++) {
            if (list.get(i) == k) {
                return false;
            }
            if (list.get(i) - i == k - list.size()) {
                return false;
            }
            if (list.get(i) + i == k + list.size()) {
                return false;
            }
        }
        return true;
    }
};


0
0

猜你在找
【直播】机器学习&数据挖掘7周实训--韦玮
【套餐】系统集成项目管理工程师顺利通关--徐朋
【直播】3小时掌握Docker最佳实战-徐西宁
【套餐】机器学习系列套餐(算法+实战)--唐宇迪
【直播】计算机视觉原理及实战--屈教授
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之矩阵--黄博士
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之凸优化--马博士
【套餐】Javascript 设计模式实战--曾亮
查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:30619次
    • 积分:1919
    • 等级:
    • 排名:千里之外
    • 原创:166篇
    • 转载:8篇
    • 译文:0篇
    • 评论:49条