POJ 1321 棋盘问题

http://poj.org/problem?id=1321

大意：在一个不规则棋盘，即‘#’处。摆放K个棋子，要求横竖不同行。

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

int map[10][10];
int n,k;
int ans;
int v[10];

void Init()
{
    int i,j;
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
            map[i][j]=1;
}

void dfs(int cur,int tot)
{
    int i;
    if(tot==k)
    {
        ans++;
        return;
    }
    if(cur>=n)//与后面的注释对应
        return;

    for(i=0;i<n;i++)
    {
        if(map[cur][i]==0 && !v[i])
        {
            v[i]=1;
            dfs(cur+1,tot+1);
            v[i]=0;
        }
    }
    dfs(cur+1,tot);    //这里要注意，因为k小于n，所以可能在最后一行之前就摆好棋子。所以还要往下搜索在下一行摆放，与之前的对应防止越界。
    return;
}

int main()
{
    char t[10];
    while(scanf("%d%d",&n,&k)!=EOF && n!=-1 && k!=-1)
    {
        getchar();

        Init();
        memset(v,0,sizeof(v));
        for(int i=0;i<n;i++)
        {
            scanf("%s",&t);

            for(int j=0;j<n;j++)
                if(t[j]=='#')map[i][j]=0;
                else map[i][j]=1;
        }
        ans=0;
        dfs(0,0);
        printf("%d\n",ans);
    }
    return 0;
}
/*
4 4
####
####
####
####
*/