POJ - 1321

题意

略

题解1(状态压缩DP)

分析

之前所用的DP数组下标都是整数的形式, 为了用DP求解, 就需要将题目所给的棋盘抽象成整数. 注意到一行棋子的可能状态可以用1<< n来表示. 而且再考虑到逐行处理的话, 就很自然地得出DP数组的定义方式.

状态定义:dp[i][j] 表示第i行, 对应状态为j时的组合数.

需要注意的是, 这里的j所对应的不一定是k个棋子. 所以要得出最后的答案, 需要扫一遍dp[n][j] , 当且仅当j对应的恰好是k个棋子, 才能加上. 这一步用bitset写的话会很简洁.

代码

#include<set>
#include<map>
#include<cmath>
#include<queue>
#include<cctype>
#include<cstdio>
#include<vector>
#include<string>
#include<sstream>
#include<cstdlib>
#include<cstring>
#include<float.h>   //X=FLT,DBL,LDBL;X_MANT_DIG,X_DIG,X_MIN_10_EXP,X_MIN_10_EXP,X_MIN,X_MAX,X_EPSILON
#include<limits.h>  //INT_MAX,LLONG_MAX
#include<iostream>
#include<algorithm>
#include<bitset>
using namespace std;
typedef long long ll;
const double PI = acos(-1.0);
#define INF 0x3f3f3f3f

char c[19][19];
int dp[19][1<<13];
int n, k;

int main(void){
    //freopen("in", "r", stdin);
    while(~scanf("%d%d", &n, &k)){
        if(n==-1&&k==-1) break;
        for(int i=1; i<=n; i++){
            getchar();
            for(int j=1; j<=n; j++)
                scanf("%c", &c[i][j]);
        }
        memset(dp, 0, sizeof dp);
        dp[0][0]=1;
        for(int i=1; i<=n; i++){
            for(int j=0; j<1<<n; j++){
                dp[i][j]+=dp[i-1][j];
                for(int l=0; l<n; l++){
                    if(c[i][l+1]=='#' && !(1<<l&j)){
                        dp[i][1<<l | j]+=dp[i-1][j];
                    }
                }
            }
        }
        int ans=0;
        for(int i=0; i<1<<n; i++){
            bitset<10> a(i);
            if(a.count()==k)
                ans+=dp[n][i];
        }
        printf("%d\n", ans);
    }

    return 0;
}

题解2(DFS）

分析

考虑到数据不大，我们可以尝试用dfs求解．这里需要在dfs函数加上两个参数，即当前到达的行数和待摆放的棋子个数．为了方便，我们可以从最后一行往前推．
需要注意的是，假如Ａ行某处存在３处可摆放棋子，那么这一行的状态转移方式有４种，即需要注意涵盖３处都不摆放这种方式．

代码

#include<set>
#include<map>
#include<ctime>     //CLOCKS_PER_SEC;clock_t t=clock();
#include<cmath>     
#include<queue>
#include<bitset>
#include<cctype>
#include<cstdio>
#include<vector>
#include<string>    //getline(cin, line);
#include<sstream>   //stringstream ss(line);(ss is a stream like cin).
#include<cstdlib>
#include<cstring>
#include<float.h>   //X=FLT,DBL,LDBL;X_MANT_DIG,X_DIG,X_MIN_10_EXP,X_MIN_10_EXP,X_MIN,X_MAX,X_EPSILON
#include<limits.h>  //INT_MAX,LLONG_MAX
#include<iostream>  //ios_base::sync_with_stdio(false);
#include<algorithm>
using namespace std;
typedef long long ll;
typedef vector<int> vi;
typedef pair<int,int> pii;
const double PI = acos(-1.0);
const int INF=0x3f3f3f3f;
const int Max=1;

char c[9][9];
int ans, n, k, vis[9];

void dfs(int row, int cnt){
    if(cnt==0){
        ans++;
        return;
    }
    if(row<0)
        return;
    for(int i=0; i<n; i++){
        if(c[row][i]=='#' && !vis[i]){
            vis[i]=1;
            dfs(row-1, cnt-1);
            vis[i]=0;
        }
    }
    dfs(row-1, cnt);
}


int main(void){
    //freopen("in", "r", stdin); freopen("out", "w", stdout);
    while(~scanf("%d%d", &n, &k)){
        if(n==-1&&k==-1) break;
        for(int i=0; i<n; i++){
            getchar();
            for(int j=0; j<n; j++)
                scanf("%c", &c[i][j]);
        }
        ans=0;
        memset(vis, 0, sizeof vis);
        dfs(n-1, k);
        printf("%d\n", ans);
    }

    return 0;
}