1. 程式人生 > >POJ-1321-棋盤問題

POJ-1321-棋盤問題

原題: 在一個給定形狀的棋盤(形狀可能是不規則的)上面擺放棋子,棋子沒有區別。要求擺放時任意的兩個棋子不能放在棋盤中的同一行或者同一列,請程式設計求解對於給定形狀和大小的棋盤,擺放k個棋子的所有可行的擺放方案C。 Input 輸入含有多組測試資料。 每組資料的第一行是兩個正整數,n k,用一個空格隔開,表示了將在一個nn的矩陣內描述棋盤,以及擺放棋子的數目。 n <= 8 , k <= n 當為-1 -1時表示輸入結束。 隨後的n行描述了棋盤的形狀:每行有n個字元,其中 # 表示棋盤區域, . 表示空白區域(資料保證不出現多餘的空白行或者空白列)。 Output 對於每一組資料,給出一行輸出,輸出擺放的方案數目C (資料保證C<2^31)。 Sample Input 2 1 #. .# 4 4 …# …#. .#… #… -1 -1 Sample Output 2 1 題意: 有一個n

n的棋盤,上面有‘#’標記的是可以放置棋子的位置,現在要求放置k顆棋子,要求每一行每一列最多隻能有一顆棋子,問有多少種方法。 題解: 這道題目以前寫過,過了很久又回來寫,也沒多想就直接暴力深搜,結果把問題搞得很複雜,回首看了一下原先寫的程式碼,回憶起來了應該如何搜尋。因為只能每一行每一列都只能有一個棋子,所以我們可以按行來深搜,搜尋每一行中有沒有可以放置的位置,依次去深搜,若是放夠了k個棋子,則方法數加一。 注意:因為需要計數,所以一每一次深搜都要回溯。(更多細節見程式碼) 附上AC程式碼:

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int n,k,cnt,num;
int dir[4][2]={1,1,1,-1,-1,-1,-1,1};
int vis[10];
char a[10][10];
void dfs(int l)
{
    if(num==k)//如果放置的棋子等於k,則方法數量加一
    {
        cnt++;
        return;
    }
    if(l>=n)//如果行數大於n就返回
        return;
    else
    {
        for(int i=0;i<n;++i)
        {
            if(a[l][i]=='#'&&!vis[i])//如果是可以放置棋子的位置並且此行沒有被標記過
            {
                vis[i]=1;
                num++;
                dfs(l+1);
                num--;//因為要計算有多少种放法,所以必須要回溯
                vis[i]=0;
            }
        }
    }
    dfs(l+1);//若是當前行沒有則直接遍歷下一行
}
int main()
{
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        memset(vis,0,sizeof(vis));//重置vis陣列,否則會WA
        num=cnt=0;//重置
        if(n==-1&&k==-1)break;
        for(int i=0;i<n;i++)
            scanf("%s",&a[i]);
        dfs(0);//從第一行開始遍歷
        printf("%d\n",cnt);
    }
    return 0;
}

歡迎評論!