POJ 1321 棋盘问题 dfs 难度:0
http://poj.org/problem?id=1321
注意是在'#'的地方放棋子
矩阵大小不过8*8,即使是8!的时间复杂度也足以承受,可以直接dfs求解
dfs时标注当前点的行和列已被访问,接着搜索行列都未被访问的新点,注意搜索完毕之后标注当前点的行和列未被访问
#include <cstdio> #include <cstring> using namespace std; int n,k; char maz[8][9]; int e[8][8],len[8]; bool r[8],c[8]; void dye(int x,int y){ r[x]=c[y]=true; } void undye(int x,int y){ r[x]=c[y]=false; } int dfs(int s,int cnt){ if(cnt<=0)return 1; int ans=0; for(int i=s+1;cnt+i<=n;i++){ for(int j=0;j<len[i];j++){ if(r[i]||c[e[i][j]])continue; dye(i,e[i][j]); ans+=dfs(i,cnt-1); undye(i,e[i][j]); } } return ans; } int main(){ while(scanf("%d%d",&n,&k)==2&&n!=-1){ for(int i=0;i<n;i++)scanf("%s",maz[i]); memset(len,0,sizeof(len)); memset(r,0,sizeof(r)); memset(c,0,sizeof(c)); for(int i=0;i<n;i++){ for(int j=0;j<n;j++){ if(maz[i][j]=='#'){ e[i][len[i]++]=j; } } } if(k==0){ puts("0"); continue; } int ans=0; for(int i=0;k+i<=n;i++){ for(int j=0;j<len[i];j++){ dye(i,e[i][j]); ans+=dfs(i,k-1); undye(i,e[i][j]); } } printf("%d\n",ans); } return 0; }