codeforces1209E2 状压dp，枚举子集

题意

给一个n行m列的矩阵，每一列可以循环移位，问经过任意次移位后每一行的最大值总和最大为多少。

分析

每一行的最大值之和最大，可以理解为每一行任意选择一个数使它们的和最大。

设\(dp[i][S]\)为前i列，已经确定值的行集合为S时，和的最大值。答案为\(dp[m][(1<<n)-1]\),

设\(f[i][S]\)为第i列循环位移任意次后选择元素所在行构成的集合为S的最大值，可以预处理出。

枚举第i+1列循环位移的次数和第i+1列选择哪些元素(枚举子集)计入答案来转移。

转移方程为\(dp[i][S]=max(dp[i-1][S']+f[i][S-S'],dp[i][S]),S'为S的子集\)。

复杂度为\(O(T*m*(2^n*n^2+3^n))\),显然过不了，实际上我们只需要考虑每列最大值最大的前n列，因为选择其他列不如替换成这n列中还没用的列。这样复杂度就能降到\(O(T*n*(2^n*n^2+3^n))\)。

Code

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define lson l,mid,p<<1
#define rson mid+1,r,p<<1|1
#define ll long long
using namespace std;
const int inf=1e9;
const int mod=1e9+7;
const int maxn=1e5+10;
int t,n,m,a[15][2010],dp[15][1<<13],r[2010],b[2010],f[15][1<<13];
int cmp(int x,int y){
    return r[x]>r[y];
}
int main(){
    //ios::sync_with_stdio(false);
    //freopen("in","r",stdin);
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        for(int i=0;i<m;i++) r[i]=0,b[i]=i;
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                scanf("%d",&a[i][j]);
                r[j]=max(r[j],a[i][j]);
            }
        }
        sort(b,b+m,cmp);
        m=min(n,m);
        for(int i=0;i<m;++i){
            for(int j=0;j<(1<<n);++j){
                int u=0;
                for(int k=0;k<n;++k){
                    int ret=0;
                    for(int l=0;l<n;++l) if((j>>l)&1) ret+=a[(l+k)%n][b[i]];
                    u=max(u,ret);
                }
                f[i][j]=dp[i][j]=u;
                if(i==0) continue;
                for(int k=j;k;k=(k-1)&j) dp[i][j]=max(dp[i-1][k]+f[i][j-k],dp[i][j]);
            }
        }
        printf("%d\n",dp[m-1][(1<<n)-1]);
    }
    return 0;
}

转载于:https://www.cnblogs.com/xyq0220/p/11541553.html