ZOJ 3777 - Problem Arrangement（状压DP）

题目:

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=5264

题意：

给出n*n的矩阵，从中选出n个数，任意两个数在不同行不同列，求出总和大于m的方案数。

思路：

因为给出的矩阵比较小（n<=13）,所以可以状压。

dp[i][s][k] 表示到第i行状态为s总和为j 的方案数。

总和j>m时当做m。

因为转移的时候i只与前一行有关，所以使用滚动数组，不然会爆内存。

还有一个优化是到第i行是状态为s时，判断s是否选过i个值了，没有的话此状态舍弃。

AC.

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;
int a[15][15], dp[3][1<<13][505];

int gcd(int a, int b)
{
    return b == 0? a: gcd(b, a%b);
}
int main()
{
    //freopen("in", "r", stdin);
    int T;
    scanf("%d", &T);
    while(T--) {
        int n, m;
        scanf("%d%d", &n, &m);
        for(int i = 0; i < n; ++i) {
            for(int j = 0; j < n; ++j)
                scanf("%d", &a[i][j]);
        }
        memset(dp, 0, sizeof(dp));
        dp[0][0][0] = 1;
        int id = 0;
        for(int i = 0; i < n; ++i) {
            id = 1-id;
            //memset(dp[id], 0, sizeof(dp[id]));

            for(int s = 0; s < (1<<n); ++s) {

                int c = 0;
                for(int t = 0; t < n; ++t) {
                    if((s&1<<t)) c++;
                }
                if(c != i) continue;

                 for(int j = 0; j < n; ++j) {
                     if((s>>j & 1) == 0) {

                       for(int k = 0; k <= m; ++k) {
                           if(dp[1-id][s][k] > 0) {

                                int cnt = k + a[i][j];
                                if(cnt > m) cnt = m;
                                dp[id][s|(1<<j)][cnt] += dp[1-id][s][k];
                                //printf("%d\n", dp[id][s|(1<<j)][cnt]);
                           }
                        }
                    }
                }
            }
        }
        int ans = dp[(n)%2][(1<<n)-1][m];
        //printf("ans = %d\n", ans);
        if(ans == 0) printf("No solution\n");
        else {
             int tmp = 1;
             for(int i = 1; i <= n; ++i) {
                 tmp *= i;
             }
             int g = gcd(ans, tmp);
             printf("%d/%d\n", tmp/g, ans/g);
        }
    }
    return 0;
}

递归写法。

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;
int dp[5000][1500], a[15][15];
int n, m;

int gcd(int aa, int ba)
{
    return ba == 0? aa: gcd(ba, aa%ba);
}
int dfs(int s, int r, int tal)
{
   
    if(dp[s][tal] >= 0) return dp[s][tal];
    if(s == (1<<n)-1) {
        if(tal >= m)
            return 1;
        //else return 0;
    }
    int res = 0;
    for(int i = 0; i < n; ++i) {
        if(!(s>>i & 1)) {
            res += dfs(s|1<<i, r+1, tal+a[r][i]);
        }
    }
    return dp[s][tal] = res;
}
int main()
{
    //freopen("in", "r", stdin);
    int T;
    scanf("%d", &T);
    while(T--) {
        scanf("%d%d", &n, &m);
        for(int i = 0; i < n; ++i) {
            for(int j = 0; j < n; ++j)
                scanf("%d", &a[i][j]);
        }
        memset(dp, -1, sizeof(dp));
        int ans = dfs(0, 0, 0);
        //printf("%d\n", ans);
        if(ans == 0) printf("No solution\n");
        else {
             int tmp = 1;
             for(int i = 1; i <= n; ++i) {
                 tmp *= i;
             }
             int g = gcd(ans, tmp);
             printf("%d/%d\n", tmp/g, ans/g);
        }
    }
    return 0;
}