把博客园图标替换成自己的图标
把博客园图标替换成自己的图标end

【状压dp】Light OJ 1011 - Marriage Ceremonies

题目传送门

题意简述:
一个 n ∗ m n*m nm的矩阵,现要从中选一些数,要求每一行每一列中都只能选一个数,求选的这些数的和的最大值。
多组数据。


算是状压dp的一种比较经典的类型了吧。
显然状压(数据范围,算最优需要知道前面的选法状态)就不多说了直接切入正题。

定义 d p [ i ] [ s ] dp[i][s] dp[i][s]为前 i i i行(选 i i i个数,根据题意,只能选 i i i个数,也就是 s s s的1的个数也为i,感觉这里浪费了一些空间,但是不知道怎么优化 管他的能过就行了 )选的列的状态为 s s s(二进制数)的最大和。
转移非常简单,从一个当前列没有被选过的地方转移取最大就可以了。

#include<queue>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
#define N 16
#define INF 0x3f3f3f3f
int dp[N][1<<N];
int n,a[N][N];
int main()
{
    int T;
    scanf("%d",&T);
    for(int cs=1;cs<=T;cs++)
    {
        printf("Case %d: ",cs);
        scanf("%d",&n);
        memset(dp,0,sizeof(dp));
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                scanf("%d",&a[i][j]);
        for(int i=0;i<n;i++)
            dp[0][1<<i]=a[0][i];
        for(int s=0;s<=(1<<n)-1;s++)//枚举子集
            for(int i=1;i<n;i++)
                for(int j=0;j<n;j++)
                {
                    if((1<<j)&s/*已经被选*/)
                        continue;
                    dp[i][s|(1<<j)]=max(dp[i][s|(1<<j)],dp[i-1][s]+a[i][j]);
                }
        printf("%d\n",dp[n-1][(1<<n)-1]);
    }
    return 0;
}

可以发现每次转移只与上一行有关,所以可以压成只有2行的滚动数组。

#include<queue>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
#define N 16
#define INF 0x3f3f3f3f
int dp[2][1<<N];
int n,a[N][N];
int main()
{
    int T;
    scanf("%d",&T);
    for(int cs=1;cs<=T;cs++)
    {
        printf("Case %d: ",cs);
        scanf("%d",&n);
        memset(dp,0,sizeof(dp));
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                scanf("%d",&a[i][j]);
        for(int i=0;i<n;i++)
            dp[0][1<<i]=a[0][i];
        for(int i=1;i<n;i++)
            for(int j=0;j<n;j++)
                for(int s=0;s<=(1<<n)-1;s++)//枚举子集
                {
                    if((1<<j)&s/*已经被选*/)
                        continue;
                    dp[i&1][s|(1<<j)]=max(dp[i&1][s|(1<<j)],dp[(i-1)&1][s]+a[i][j]);
                }
        printf("%d\n",dp[(n-1)&1][(1<<n)-1]);
    }
    return 0;
}
posted @ 2019-08-10 19:15  Starlight_Glimmer  阅读(9)  评论(0编辑  收藏  举报  来源
浏览器标题切换
浏览器标题切换end