leetCode练习（51）

最新推荐文章于 2024-08-28 21:02:53 发布

碰碰猪

最新推荐文章于 2024-08-28 21:02:53 发布

阅读量264

点赞数

分类专栏： leetCode 文章标签：回溯法 backtracking leetcode java

本文链接：https://blog.csdn.net/u010771890/article/details/52763799

版权

leetCode 专栏收录该内容

150 篇文章 0 订阅

订阅专栏

题目：N-Queens

难度：hard

问题描述：

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.."]
]

解题思路：著名的8皇后问题，要求找到所可行的位置，让所有皇后不会直接攻击到其他皇后。使用回溯法以找到所有的解。基本逻辑为：

1.所有皇后是否已经放置若放置成功，则将棋盘加入链表，否则到2。

2.查看已经走了几步steps，则在第steps行开始从第0列到n列以此放置新的皇后，放置前判断是否可以放置（未被其他已经存在的皇后攻击），可以的话，则放置新后，返回1.

这里的回溯法注意，每次迭代，不是从下一格开始放置新后，而是从新的一行开始放置新后，这样复杂度从O（n*n）降低到了O（n），从而可以通过提交，否则的话会超时。

具体代码如下：

public class h_51_NQueens {
	public  ArrayList<List<String>> lists=new ArrayList<List<String>>();
	public  LinkedList<String> temp;
	public  List<List<String>> solveNQueens(int n) {	
		if(n==1){
			temp=new LinkedList<String>();
			temp.add("Q");
			lists.add(temp);
			return lists;
		}
		
        char[][] board=new char[n][n];
        for(int i=0;i<n;i++){
        	for(int j=0;j<n;j++){
        		board[i][j]='.';
        	}
        }
        diedai(board,0);  
        return lists;
    }
	public  void diedai(char[][] b,int steps){	
		if(steps==b.length){//所有皇后放完，将棋盘存入lists
			lists.add(boardToList(b));
			return;
		}	
		char[][] boardtemp=new char[b.length][b.length];
		for(int ii=0;ii<b.length;ii++){
			for(int jj=0;jj<b.length;jj++){
				boardtemp[ii][jj]=b[ii][jj];
			}
		}
		for(int n=0;n<b.length;n++){
			if(!this.isvalidate(b, steps, n)){
				continue;
			}else{
				boardtemp[steps][n]='Q';
				diedai(boardtemp,steps+1);
			}
			boardtemp[steps][n]='.';
		}
		
	}
	public boolean isvalidate(char[][] b,int x,int y){
		for(int i=0;i<b.length;i++){
			for(int j=0;j<b.length;j++){
				if(b[i][j]=='Q'&&(x==i||y==j||(Math.abs(i-x)==Math.abs(j-y)))){
					return false;
				}
			}
		}
		return true;
	}
	public  List<String> boardToList(char[][]b){//将棋盘转为list<String>
		String str;
		temp=new LinkedList<String>();
		for(int i=0;i<b.length;i++){
			str=new String(b[i]);
			temp.add(str);
		}
		System.out.println("solution:");
		for(int i=0;i<temp.size();i++){
			System.out.println(temp.get(i));
		}
		return temp;
	}
	
	public static void main(String[]args){
		h_51_NQueens s=new h_51_NQueens();
		s.solveNQueens(4);
	}
}