poj 1321 棋盘问题 dfs暴力

淡淡的烟草味﹌ 2022-08-18 11:57 179阅读 0赞

棋盘问题

<table> 
 <tbody> 
  <tr> 
   <td><strong>Time Limit:</strong> 1000MS</td> 
   <td>&nbsp;</td> 
   <td><strong>Memory Limit:</strong> 10000K</td> 
  </tr> 
  <tr> 
   <td><strong>Total Submissions:</strong> 30247</td> 
   <td>&nbsp;</td> 
   <td><strong>Accepted:</strong> 14980</td> 
  </tr> 
 </tbody> 
</table>

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

题意...........

分析: dfs暴力,对于每一行,有两种选择,可以选择在某一列放置一个棋子,或者不放置棋子.参考了别人的代码.

#include<bitset> //32ms
    #include<map>
    #include<vector>
    #include<cstdio>
    #include<iostream>
    #include<cstring>
    #include<string>
    #include<algorithm>
    #include<cmath>
    #include<stack>
    #include<queue>
    #include<set>
    #define inf 0x3f3f3f3f
    #define mem(a,x) memset(a,x,sizeof(a))
    
    using namespace std;
    
    typedef long long ll;
    typedef pair<int,int> pii;
    
    inline int in()
    {
        int res=0;char c;
        while((c=getchar())<'0' || c>'9');
        while(c>='0' && c<='9')res=res*10+c-'0',c=getchar();
        return res;
    }
    const int N=100100;
    
    
    char a[10][10];
    bool vis[10];
    
    void dfs(int row,int num,int &ans,int n,int m)//行数，棋子个数
    {
        if(num==m)
        {
            ans++;
            return;
        }
        if(row >= n) return;
        
        //当前的行放或者不放棋子
        for(int i=0;i<n;i++) //枚举每一列，在某一列放置一个棋子
        {
            if(a[row][i] == '#' && !vis[i])
            {
                vis[i]=1;
                dfs(row+1,num+1,ans,n,m);//当前的行放了一个棋子,进入下一行
                vis[i]=0;                 //回溯的时候要把标记取消了,因为要在下一列放置棋子
            }
        }
        dfs(row+1,num,ans,n,m);//当前的行都不放棋子
    }
    int main()
    {
        int n,m;
        while(~scanf("%d%d",&n,&m))
        {
            if(n==-1) return 0;
            for(int i=0;i<n;i++) scanf("%s",a[i]);
            mem(vis,0);
            int ans=0;
            dfs(0,0,ans,n,m);
            printf("%d\n",ans);
        }
        return 0;
    }