POJ 1321 dfs

在棋盘上放置棋子使它们任意两个都不在同一行或同一列

dfs（x，k）表示的是访问到第x行已放置了k个棋子

这道题我们以每行或者每列为单位来看题，每次搜索都对一整行进行访问，并在安置棋子的点的列位置上使其visit[col]=1

如果放置成功那么dfs（x+1，k+1），再进行回溯

不论是否成功，都要dfs（x+1，k），确保这个位置我不放棋子也曾进行访问过

 

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
int mat[9][9],dp[9][9],visit[9],n,k,cnt;

int dfs(int x,int a)
{
    //if(dp[x][a]) return dp[x][a];这句话写了就错了
    if(a==k) return 1;
    if(x>n) return 0;

    int nowstate=0;

    for(int i=1;i<=n;i++){
        if(mat[x][i]&&!visit[i]){
            visit[i]=1;
            nowstate+=dfs(x+1,a+1);
            visit[i]=0;
        }
    }

    nowstate+=dfs(x+1,a);
    dp[x][a]=nowstate;
    return nowstate;
}
int main()
{
    char a;
    while(scanf("%d%d",&n,&k)){

        memset(visit,0,sizeof(visit));
        memset(dp,0,sizeof(dp));
        cnt=0;

        if(n==-1&&k==-1)
            break;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++){
                cin>>a;
                mat[i][j]=a=='#'?1:0;
            }

        int ans=dfs(1,0);
        printf("%d\n",ans);
    }
    return 0;
}