Leetcode 51. N-Queens

不定期更新leetcode解题java答案。

采用pick one的方式选择题目。

本题的意思是采用国际象棋的方式添加棋子“后”，使之相互间不在自己的攻击范围内。“后”的攻击方式为横方向、纵方向以及斜方向均可攻击。

使用傻瓜式方法，逐行进行尝试，循环，回溯。得出最后可能出现的结果。直接上代码：

public class Solution {
    public List<List<String>> solveNQueens(int n) {
        List<List<String>> result = new ArrayList();
        
        StringBuilder sb = new StringBuilder();
        for(int i = 0; i < n; i++)
            sb.append(",");
        String init = sb.toString();
        
        Queue<List<String>> q = new LinkedList();
        //initialize first line
        for(int i = 0; i < n; i++){
            List<String> list = new ArrayList();
            for(int j = 0; j < n; j++)
                list.add(j == 0 ? addQueen(i, n) : init);
            attackPlace(list, 0, n);
            q.offer(list);
        }
        //add queen to each line except first line
        for(int i = 1; i < n; i++){
            int size = q.size();
            for(int j = 0; j < size; j++){
                List<String> list = q.poll();
                for(int k = 0; k < n; k++){
                    if(list.get(i).charAt(k) != '.'){
                        List<String> tmp = new ArrayList();
                        for(int s = 0; s < n; s++)
                            tmp.add(list.get(s));
                        tmp.set(i, addQueen(k, n));
                        attackPlace(tmp, i, n);
                        q.offer(tmp);
                    }
                }
            }
        }
        
        int size = q.size();
        for(int i = 0; i < size; i++)
            result.add(q.poll());
        return result;
    }
    
    public void attackPlace(List<String> list, int line, int size){
        int loc = list.get(line).indexOf("Q");
        
        for(int i = line + 1; i < size; i++){
            StringBuilder str = new StringBuilder(list.get(i));
            if(loc + i - line < size)
                str.replace(loc + i - line, loc + i - line + 1, ".");
            if(loc - i + line >= 0)
                str.replace(loc - i + line, loc - i + line + 1, ".");
            str.replace(loc, loc + 1, ".");
            list.set(i, str.toString());
        }
    }
    
    public String addQueen(int loc, int size){
        StringBuilder str = new StringBuilder();
        for(int s = 0; s < size; s++)
            if(s == loc)
                str.append("Q");
            else
                str.append(".");
        return str.toString();
    }
}

初始以“,”视为可放置“后”的位置，在每放置一个“后”对后面“,”使用题目要求“.”进行替换。函数attackPlace进行此操作。

addQueen函数单纯的是在添加一个“后”将此行的所有位置视为不可添加的符号替换，稍微提高点运行的速度。

由于本人对效率研究不是很多，最初的提交都是TLE，此代码运行速度也不是很快，如果有提高效率的方法或者具体的细节也欢迎来交流。本文仅提供一种解决思路。