poj 1321 棋盤問題 (dfs)

標籤：dfs   poj   

連結:poj 1321

題意：給定n*n的棋盤， # 代表棋盤地區，可以放棋子, 

      .代表空白地區（不能放棋子），現要擺放k個棋子，求方案數.

要求:擺放時任意的兩個棋子不能放在棋盤中的同一行或者同一列

思路：因為任意兩個棋子不能在同一行或同一列，可以逐行dfs，並對已訪問的行標記

#include<stdio.h>#include<string.h>char s[10][10];__int64 cnt;int sum,n,m,vis[10];void dfs(int r){    int i,j;    for(i=r+1;i<n;i++)  //從該行往下dfs        for(j=0;j<n;j++)            if(s[i][j]=='#'&&!vis[j]){                  sum++;                vis[j]=1;     //訪問後標記為1                if(sum==m)                    cnt++;                else                    dfs(i);                sum--;     //回溯                vis[j]=0;            }}int main(){    int i,j;    while(scanf("%d%d",&n,&m)!=EOF){        if(m==-1&&n==-1)            break;        cnt=0;        for(i=0;i<n;i++)            scanf("%s",s[i]);        for(i=0;i<n;i++)            for(j=0;j<n;j++){                memset(vis,0,sizeof(vis));    //先初始化為0                if(s[i][j]=='#'){      //每找到一個可以放棋子的地方，就進行dfs                    sum=1;                    vis[j]=1;                    if(sum==m){                        cnt++;                        continue;                    }                    dfs(i);                }            }        printf("%I64d\n",cnt);    }    return 0;}

poj 1321 棋盤問題 (dfs)