POJ 1321 棋盤問題（DFS & 狀壓DP）

    用DFS寫固然很簡單了，8！的複雜度，16MS搞定。

    在Discuss裏看到有同窗用狀態壓縮DP來寫，就學習了一下，果真很精妙呀。

    狀態轉移分兩種，當前行不加棋子，和加棋子。dp[i][j]中，i表明行數，j表明當前行棋子的狀態。j的二進制中，1表明有旗子，0表明無棋子。

    貼代碼~狀壓DP果真快一點。

#include <cstdio>
#include <cstring>

int n,k,count;
bool mp[10][10];
int num[256];
int dp[9][256];

int main()
{
//    freopen("in.txt","r",stdin);

    for(int i=1;i<256;i++)
    {
        int tmp=i;
        while(tmp)
        {
            if(tmp&1)
                num[i]++;
            tmp>>=1;
        }
    }

    while(~scanf("%d%d",&n,&k) && n!=-1 && k!=-1)
    {
        char str[20];
        for(int i=1;i<=n;i++)
        {
            scanf("%s",str+1);
            for(int l=1;l<=n;l++)
            {
                if(str[l]=='#')
                    mp[i][l]=true;
                else
                    mp[i][l]=false;
            }
        }

        int status=1<<n;

        memset(dp,0,sizeof(dp));
        dp[0][0]=1;
        for(int i=1;i<=n;i++)
        {
            for(int j=0;j<status;j++) if(num[j]<=k)
            {
                dp[i][j]+=dp[i-1][j];
                for(int l=1;l<=n;l++) if(mp[i][l] && (j&(1<<(l-1)))==0)
                {
                    dp[i][(j|(1<<(l-1)))]+=dp[i-1][j];
                }
            }
        }

        int ans=0;
        for(int i=0;i<status;i++) if(num[i]==k)
            ans+=dp[n][i];

        printf("%d\n",ans);
    }
}

    還有傳統的DFS……

#include <cstdio>
#include <cstring>

int n,k,count;
bool mp[10][10];
bool col[10];

void DFS(int x,int rest)
{
    if(rest==0)
    {
        count++;
        return;
    }
    if(x>n)
        return;
    for(int i=1;i<=n;i++) if(!col[i] && mp[x][i])
    {
        col[i]=true;
        DFS(x+1,rest-1);
        col[i]=false;
    }
    if(rest+x<=n)
        DFS(x+1,rest);
}

int main()
{
//  freopen("in.txt","r",stdin);
    while(~scanf("%d%d",&n,&k) && n!=-1 && k!=-1)
    {
        memset(col,0,sizeof(col));
        char str[20];
        for(int i=1;i<=n;i++)
        {
            scanf("%s",str+1);
            for(int k=1;k<=n;k++)
            {
                if(str[k]=='#')
                    mp[i][k]=true;
                else
                    mp[i][k]=false;
            }
        }

        count=0;
        DFS(1,k);
        printf("%d\n",count);
    }
}