poj1321 DFS

 

 

 

        如题：http://poj.org/problem?id=1321

 

 

       注意这句话“其中 # 表示棋盘区域”，一开始我以为棋子只能放在“.”的位置，没想到只能放在"#"的位置才对....无语。。。。汉字题目都没看懂

 

       vis[i][j]=2代表这一列有其它棋子，不能放，vis[i][j]=1代表是“.”，所以不能放..其实只要不能放就设为已经访问过(1)就是可以的，我只是为了输出路径查错误。

 

     按行搜索就行，思路：从第一行开始，每一行最多只能放1个或者不放，放下后再在下一行找一个可以放的位置放。

    有一个剪枝，n-row<k-num时，代表剩下的(n-row)行，就算每一行都放,剩余的棋子(k-num)也比它多，直接return。

     

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;

char map[10][10];
int n,k;
int vis[10][10];

int count;
void dfs(int row,int num) //搜索到当前行，已放的棋子数目
{
 if(num==k)
 {
  count++;
 /* int j,k;
  for(j=0;j<n;j++)
   for(k=0;k<n;k++)
    if(vis[j][k]==1&&map[j][k]=='#')
     printf("(%d,%d) ",j,k);
  printf("\n");*/
  return;
 }
 if(row==n||n-row<k-num)
  return ;
 int i,j;
 for(i=0;i<n;i++)
  if(!vis[row][i]&&map[row][i]=='#')
  {
   vis[row][i]=1;
   for(j=row+1;j<n;j++)
    vis[j][i]=2;    
   dfs(row+1,num+1);
   for(j=row;j<n;j++)
    vis[j][i]=0;
   //vis[row+1][i]=0;
  }
 dfs(row+1,num);

}
int main()
{
// freopen("C:\\1.txt","r",stdin);
 while(scanf("%d%d",&n,&k)&&n!=-1||k!=-1)
 {
  memset(vis,1,sizeof(vis));
  int i,j;
  getchar();
  for(i=0;i<n;i++)
  {
   for(j=0;j<n;j++)
   {
    scanf("%c",&map[i][j]);
    if(map[i][j]=='#')
     vis[i][j]=0;
   }
   getchar();
  }
  count=0;
  dfs(0,0);
  printf("%d\n",count);
 }
 return 0;
}