码迷,mamicode.com
首页 > 其他好文 > 详细

[LeetCode] NQueen II

时间:2014-09-03 01:29:25      阅读:234      评论:0      收藏:0      [点我收藏+]

标签:style   blog   color   os   io   ar   for   div   sp   

public class Solution {
    public List<String[]> solveNQueens(int n) {
        int [] locations;
        locations = new int[n];
        List<String[]> result = new ArrayList<String[]>();

        getNQueens(locations, 0, result);
        return result;
    }
    
    public void getNQueens(int [] locations, int row, List<String[]> solutions) {
        if (row == locations.length) {
            String [] solution = new String[row];
            for(int i=0; i<locations.length; i++) {
                StringBuffer tmpRow = new StringBuffer();
                for (int j=0; j<locations.length; j++) {
                    if(j == locations[i]) tmpRow.append("Q");
                    else tmpRow.append(".");
                }
                solution[i] = tmpRow.toString();
            }
            solutions.add(solution);
        }

        for (int i=0; i<locations.length; i++) {
            if (check(locations, row, i)) {
                locations[row] = i;
                getNQueens(locations, row+1, solutions);
            }
        }
    }
    
    public boolean check(int [] locations, int row, int clom) {
        for (int i=0; i<row; i++) {
            if (locations[i] == clom) return false;
            if (locations[i]+i == row+clom) return false;
            if (i-locations[i] == row-clom) return false;
        }

        return true;
    }
}

 

[LeetCode] NQueen II

标签:style   blog   color   os   io   ar   for   div   sp   

原文地址:http://www.cnblogs.com/yuhaos/p/3952656.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!