【回溯篇】【困难】leetcode第51题 n皇后问题

题目

n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。

给你一个整数 n ，返回所有不同的 n 皇后问题 的解决方案。

每一种解法包含一个不同的 n 皇后问题 的棋子放置方案，该方案中 'Q' 和 '.' 分别代表了皇后和空位。

示例 1：

输入：n = 4
输出：[[".Q..","...Q","Q...","..Q."],["..Q.","Q...","...Q",".Q.."]]
解释：如上图所示，4 皇后问题存在两个不同的解法。

示例 2：

输入：n = 1
输出：[["Q"]]

提示：

1 <= n <= 9
皇后彼此不能相互攻击，也就是说：任何两个皇后都不能处于同一条横行、纵行或斜线上。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/n-queens
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

//回溯法求解， 递归寻找可能的棋盘，然后将棋盘更新为进入下一层递归的棋盘的函数rebuild，同时也需要一个能回到上一层棋盘状态的recall。 
void recurSive(int** board,int line,int n,int* returnSize,int** returnColumnSizes,char*** ans){
	int row=0,rloc=0;
	while(row<n){
			while(board[line][row]==0&&row<n){
				row++;
			}
            if(row!=n){
		    	board[line][row]=1;//表示放置皇后
				if(line==n){
				    printBoard(board,n,returnSize,returnColumnSizes,ans);
			    } 
			    else{
				    rloc=row;
				    rebuild(board,line,rloc,n);
				    recurSive(board,line+1,n,returnSize,returnColumnSizes,ans);
                    recall(board,line,rloc,n);
			    }    
			}
			row++;
		}
	return;
}
void rebuild(int** board,int line,int rloc,int n){
	int i,j;
	for(i=line,j=rloc;i<n&&j>=0;){
		board[i++][j--]=2;//表示会被其他女王攻击 
	}
	for(i=line,j=rloc;i<n&&j<n;){
		board[i++][j++]=2;
	}
}
void recall(int** board,int line,int rloc,int n){
	int i,j;
	if(rloc==0){
		return;
	}
	else{
		for(i=line,j=rloc;i<n&&j>=0;){
		board[i++][j--]=0;//表示空位 
	   }
	    for(i=line,j=rloc;i<n&&j<n;){
		board[i++][j++]=0;
	   }
    }
}
void printBoard(int** board,int n,int* returnSize,int** returnColumnSizes,char*** ans){
	int i,j;
	for(i=0;i<n;i++){
		for(j=0;j<n+1;j++){
			if(j=n){
				returnColumnSizes[*returnSize]=n*(n+1);
				ans[i][j][(*returnSize)++]='\0';
			}
			else{
				if(board[i][j]==1){
					ans[i][j][*returnSize]='Q';
				}
				else{
					ans[i][j][*returnSize]='.';
				}
			}
		}
	}	
} 
//主函数
int ***ans,**board; 

returnSize=0;
ans=(char***)calloc(n*(n+1)*n,sizeof(char));
board=(int**)calloc(n*n,sizeof(int));
returnColumnSizes=(int**)calloc(n*n,sizeof(int));
recurSive(board,0,n,returnSize,returnColumnSizes,ans);

return ans;