POJ 1321(DFS)

水题。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<map>
#include<queue>
using namespace std;
#define inf 0x3f3f3f3f
#define ll long long
const int maxn=200005;
const double eps=1e-8;
const double PI = acos(-1.0);
int g[10][10],vis[10][10],cnt=0,n,k;
bool check(int r,int c)
{
    if(!g[r][c])
        return false;
    for(int i=1;i<=r;i++)
    {
        if(vis[i][c])
        return false;
    }
    return true;
}
void dfs(int r,int num)
{
    if(num==k)
    {
        cnt++;
        return ;
    }
    if(r>n)
        return ;
    int i;
    for( i=1;i<=n;i++)
    {
        if(check(r,i))
        {
            vis[r][i]=1;
            dfs(r+1,num+1);
            vis[r][i]=0;
        }
    }
    dfs(r+1,num);
}
int main()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(0);
    std::cout.tie(0);
    while(cin>>n>>k)
    {
        cnt=0;
        memset(g,0,sizeof(g));
        memset(vis,0,sizeof(vis));
        if(n==-1&&k==-1)
        {
            break;
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                char c;
                cin>>c;
                if(c=='#')
                {
                    g[i][j]=1;
                }
                else
                    g[i][j]=0;
            }
        }
        dfs(1,0);
        cout<<cnt<<endl;
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Dilly__dally/article/details/81490710