51. N-Queens

The n-queens puzzle is the problem of placing n queens on an n×n chessboard such that no two queens attack each other.

Given an integer n, return all distinct solutions to the n-queens puzzle.

Each solution contains a distinct board configuration of the n-queens' placement, where 'Q' and '.' both indicate a queen and an empty space respectively.

For example,
There exist two distinct solutions to the 4-queens puzzle:

[
 [".Q..",  // Solution 1
  "...Q",
  "Q...",
  "..Q."],

 ["..Q.",  // Solution 2
  "Q...",
  "...Q",
  ".Q.."]
]
这是著名的n皇后问题,在n*n的棋盘上摆放n个皇后,要求任意2个皇后不能在同一行、同一列以及同一斜线上,求所有满足要求的集合。Q代表皇后, .代码空格。
这个题可以用回溯的思想解决,
我们先在棋盘的第一行任意选择一个方块放置一个皇后,然后在第二行选择一个 合适的位置(指该位置所在的行、列以及斜线都没有皇后) 放置第二个皇后,依次放置下去,如果某一行的所有位置都无法放置皇后,那么回溯,改变上一个皇后的位置,再重复此过程,直到第n行也放置了一个皇后,我们将这种合理的放置方式存储在集合中,继续回溯。
下面是代码:
public class Solution {
    public List<List<String>> solveNQueens(int n) {
        char[][] ch = new char[n][n];
        for(char[] c : ch){                  //定义一个二维char数组,数组元素为'.';
            Arrays.fill(c,'.');
        }
        
        List<List<String>> list = new ArrayList<>();
        fon(ch,0,list);
        return list;  
    }
    public void fon(char[][] ch, int n,  List<List<String>> list){
        if(n == ch.length){                            //放置方案满足要求,存放在list中
            List<String> l = new ArrayList<>();
            for(int i=0;i<ch.length;i++){
                l.add(new String(ch[i]));
            }
            list.add(l);
            return;
        }else{
            for(int i=0;i<ch.length;i++){         
                if(fon1(ch,n,i)){            //在第n行的第i个元素的位置上放置一个皇后
                    fon(ch,n+1,list);       //向下递归
                    ch[n][i] = '.';         //更改回来
                }
            }   
        }
    }
    public boolean fon1(char[][] ch, int a, int b){        //判断char[a][b]是否可以放置一个皇后
        for(int i=0;i<ch.length;i++){
            if(ch[a][i] == 'Q' || ch[i][b] == 'Q')       //判断该元素所在行或者列是否已经有皇后
                return false;
            if(a+b-i>=0 && a+b-i<ch.length)               //判断该元素所在的左下到右上的斜线是否已经有皇后
                if(ch[i][a+b-i] == 'Q')
                    return false;
            if(b-a+i>=0 && b-a+i<ch.length)              //判断该元素所在的左上到右下的斜线是否已经有皇后

                if(ch[i][b-a+i] == 'Q')
                    return false;
        }
        ch[a][b] = 'Q';                       //在该位置上放置一个皇后      
        return true;
    }
}
这个方法在LeetCode上超过了83.36%的其他方法,还算不错。这是一个经典的回溯的问题,在上面的代码中,我们借用了数组,每一行的元素都可以是回溯算法中的“一条岔路”,我们使用for循环遍历所有岔路,更普遍的是使用一个集合存储所有的“岔路”,进而遍历集合。



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是使用ABT算法求解四皇后问题的Python代码: ``` def abt_search(csp): return abt_recursive(csp, {}) def abt_recursive(csp, assignment): if len(assignment) == len(csp.variables): return assignment var = select_unassigned_variable(csp, assignment) for value in order_domain_values(var, assignment, csp): if is_consistent(var, value, assignment, csp): csp.assign(var, value) inferences = {} for neighbor in csp.neighbors(var): if neighbor not in assignment: for removed_value in csp.curr_domain_values(neighbor): if not csp.constraints(var, value, neighbor, removed_value): inferences[neighbor] = removed_value csp.eliminate(neighbor, removed_value) result = abt_recursive(csp, assignment) if result is not None: return result csp.unassign(var) for neighbor, removed_value in inferences.items(): csp.revise(neighbor, removed_value) return None def select_unassigned_variable(csp, assignment): for var in csp.variables: if var not in assignment: return var def order_domain_values(var, assignment, csp): return csp.curr_domain_values(var) def is_consistent(var, value, assignment, csp): for neighbor in csp.neighbors(var): if neighbor in assignment: if not csp.constraints(var, value, neighbor, assignment[neighbor]): return False return True class CSP: def __init__(self, variables, domains, constraints): self.variables = variables self.domains = domains self.constraints = constraints self.unassigned_vars = set(variables) self.curr_domains = {v: set(domains) for v in variables} def assign(self, var, value): self.unassigned_vars.remove(var) self.curr_domains[var] = set([value]) def unassign(self, var): self.unassigned_vars.add(var) self.curr_domains[var] = set(self.domains) def eliminate(self, var, value): self.curr_domains[var].remove(value) def revise(self, var, value): self.curr_domains[var].add(value) def curr_domain_values(self, var): return self.curr_domains[var] def neighbors(self, var): return set(self.variables) - set([var]) def is_complete(self, assignment): return len(assignment) == len(self.variables) def four_queens_csp(): variables = ['Q1', 'Q2', 'Q3', 'Q4'] domains = range(1, 5) constraints = lambda Q1, Q2, Q3, Q4: (Q1 != Q2 and Q1 != Q3 and Q1 != Q4 and Q2 != Q3 and Q2 != Q4 and Q3 != Q4 and abs(Q1 - Q2) != 1 and abs(Q1 - Q3) != 2 and abs(Q1 - Q4) != 3 and abs(Q2 - Q3) != 1 and abs(Q2 - Q4) != 2 and abs(Q3 - Q4) != 1) return CSP(variables, domains, constraints) if __name__ == '__main__': csp = four_queens_csp() solution = abt_search(csp) if solution is not None: print(solution) else: print('No solution found') ``` 该代码定义了一个CSP类,其中包含“assign”、“unassign”、“eliminate”和“revise”等方法,用于实现ABT算法。它还定义了一个“four_queens_csp”函数,用于创建四皇后问题的CSP实例。最后,它调用“abt_search”函数来解决问题,并输出解决方案。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值