POJ 1321 dfs

標籤：style   blog   color   os   io   ar   for   div   sp   

在棋盤上放置棋子使它們任意兩個都不在同一行或同一列

dfs（x，k）表示的是訪問到第x行已放置了k個棋子

這道題我們以每行或者每列為單位來看題，每次搜尋都對一整行進行訪問，並在安置棋子的點的列位置上使其visit[col]=1

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

不論是否成功，都要dfs（x+1，k），確保這個位置我不放棋子也曾進行訪問過

 

 1 #include <cstdio> 2 #include <cstring> 3 #include <iostream> 4 using namespace std; 5 int mat[9][9],dp[9][9],visit[9],n,k,cnt; 6  7 int dfs(int x,int a) 8 { 9     //if(dp[x][a]) return dp[x][a];這句話寫了就錯了10     if(a==k) return 1;11     if(x>n) return 0;12 13     int nowstate=0;14 15     for(int i=1;i<=n;i++){16         if(mat[x][i]&&!visit[i]){17             visit[i]=1;18             nowstate+=dfs(x+1,a+1);19             visit[i]=0;20         }21     }22 23     nowstate+=dfs(x+1,a);24     dp[x][a]=nowstate;25     return nowstate;26 }27 int main()28 {29     char a;30     while(scanf("%d%d",&n,&k)){31 32         memset(visit,0,sizeof(visit));33         memset(dp,0,sizeof(dp));34         cnt=0;35 36         if(n==-1&&k==-1)37             break;38         for(int i=1;i<=n;i++)39             for(int j=1;j<=n;j++){40                 cin>>a;41                 mat[i][j]=a==‘#‘?1:0;42             }43 44         int ans=dfs(1,0);45         printf("%d\n",ans);46     }47     return 0;48 }

 

POJ 1321 dfs