LeetCode N-Queens II

题目

Follow up for N-Queens problem.

Now, instead outputting board configurations, return the total number of distinct solutions.

 

和上一题类似，只需要求解解的数量，简化一下就可以了，具体见上一题。

 

代码：

class Solution {
	int ans;		//结果
	vector<int> frows,fcols,frowcol,fcolrow;	//相应行有Q,列有Q，右下方向斜线有Q，右上方向斜线有Q
public:
    bool isValid(int row,int col,int n)	//判断相应位置是否可以插入Q
	{	
		if(frows[row]==1||fcols[col]==1)	//行有，列有Q
			return false;
		if(frowcol[row-col+n-1]==1)	//线x-y+n-1=i有Q
			return false;
		if(fcolrow[row+col]==1)	//线x+y=i有Q
			return false;
		return true;
	}
	void innerSolve(int row,int col,int n,int num)	//行、列、n、已经插入的Q数
	{
		if(num==n)
			ans++;
		else	//寻找之后所有可以插入的位置
		{
			int i,j;
			for(i=row;i<n;i++)
			{
				for(j=0;j<n;j++)
				{
					if(isValid(i,j,n))
					{
						frows[i]=1;
						fcols[j]=1;
						frowcol[i-j+n-1]=1;
						fcolrow[i+j]=1;
						innerSolve(i+1,0,n,num+1);	//递归求解
						frows[i]=0;
						fcols[j]=0;
						frowcol[i-j+n-1]=0;
						fcolrow[i+j]=0;
					}
				}
			}
		}
	}
	int totalNQueens(int n){
		ans=0;
		frows.clear();
		fcols.clear();
		frowcol.clear();
		fcolrow.clear();
		int i;
		for(i=0;i<n;i++)
		{
			frows.push_back(0);
			fcols.push_back(0);
		}
		for(i=0;i<2*n;i++)
		{
			frowcol.push_back(0);
			fcolrow.push_back(0);
		}
		innerSolve(0,0,n,0);	//求解
		return ans;
	}
};