您的位置:首页 > 其它

POJ 1321 棋盘问题 dfs

2017-04-06 17:24 387 查看
棋盘问题
Time Limit: 1000MS	 	Memory Limit: 10000K
Total Submissions: 24577	 	Accepted: 12150
Description
在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,请编程求解对于给定形状和大小的棋盘,摆放k个棋子的所有可行的摆放方案C。
Input
输入含有多组测试数据。
每组数据的第一行是两个正整数,n k,用一个空格隔开,表示了将在一个n*n的矩阵内描述棋盘,以及摆放棋子的数目。 n <= 8 , k <= n
当为-1 -1时表示输入结束。
随后的n行描述了棋盘的形状:每行有n个字符,其中 # 表示棋盘区域, . 表示空白区域(数据保证不出现多余的空白行或者空白列)。
Output
对于每一组数据,给出一行输出,输出摆放的方案数目C (数据保证C<2^31)。
Sample Input
2 1
#.
.#
4 4
...#
..#.
.#..
#...
-1 -1
Sample Output
2
1

//这题开始我想用类似于N皇后迭代的方法去做,可是WA

#include<iostream>
using namespace std;

char a[100][100];
int PosR[100];//记录第k行存放在第几列
int n,k;
double count;

int ChessBoard(int x)
{
if(x==k)//已经摆好了k个棋子
{
count++;
}
int i,j;
for( i=0;i<n;i++)//0~k-1行摆好,在摆第k行
{
for( j=0;j<x;j++)//和前面摆好的比较,看是否有冲突
{
if(a[i][j]=='#')//可放棋子
{
if(PosR[j]==i)
break;
}
}
if(j==x)//没有break,说明i位置可行
{
PosR[j]=i;
ChessBoard(x+1);
}
}
return count;
}

int main()
{
while(cin>>n>>k)
{
count=0;
if(n==-1&&k==-1)
break;
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
cin>>a[i][j];
}
}
cout<<ChessBoard(0)<<endl;
}
return 0;
}

//这是百度后自己写的,但我一直理解不了dfs函数中if语句中最后那两句,后来手工模拟一遍才知道其中的含义,其实是回退后继续同行搜索
#include<iostream>
#include<cstring>
using namespace std;

char ChessBoard[100][100];
int vis[100];//记录访问某列
int count;//统计摆放的棋子数
int sum;//统计方案数
int n,k;

void Dfs(int x)
{
if(count==k)
{
sum++;
return;
}
if(x>n)
return;
for(int i=0;i<n;i++)
{
if(!vis[i]&&ChessBoard[x][i]=='#')
{//满足条件
count++;
vis[i]=1;
Dfs(x+1);
count--;//类似于下棋时的悔棋,继续看同行是否有新路径(可能是递归一遍了,也可能遇到了冲突)
vis[i]=0;
}
}
Dfs(x+1);//这层没有符合条件的了进入下一层
}
int main()
{
while(cin>>n>>k)
{
count=sum=0;
if(n==-1&&k==-1)
break;
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
cin>>ChessBoard[i][j];
}
}
memset(vis,0,sizeof(vis));
Dfs(0);
cout<<sum<<endl;
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  POJ 1321 棋盘问题 dfs