【DP】 LightOJ 1011 - Marriage Ceremonies状压

点击打开链接

每一行每一列只能取一个数字

求sum最大

先预处理下每个0-(1<<16)  按1的个数分类

然后状压

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <string>
#include <iostream>
#include <algorithm>
#include <sstream>
#include <cmath>
using namespace std;
#include <queue>
#include <stack>
#include <set>
#include <vector>
#include <deque>
#include <map>
#define cler(arr, val)    memset(arr, val, sizeof(arr))
#pragma comment(linker, "/STACK:102400000,102400000")
typedef long long  LL;
const int MAXN = 230+1;
const int MAXM = 140000;
const int INF = 0x3f3f3f3f;
const int mod = 1000000007;
vector<int>x[17];
int a[19][19];
int dp[20][1<<16];
void init()
{
    for(int i=0;i<16;i++) x[i].clear();
    for(int i=0;i<(1<<16);i++)
    {
        int q=i,sum=0;
        while(q)
        {
            if(q%2) sum++;
            q/=2;
        }
        x[sum].push_back(i);
    }
}
int main()
{
#ifndef ONLINE_JUDGE
       freopen("in.txt", "r", stdin);
   //  freopen("out.txt", "w", stdout);
#endif
    int n,t,k,cas=1;
    init();
    cin>>t;
    while(t--)
    {
        cler(dp,0);
        cin>>n;
        for(int i=1;i<=n;i++)
            for(int j=0;j<n;j++)
                cin>>a[i][j];
        for(int i=1;i<=n;i++)
        {
            int v;
            for(int j=0;j<x[i-1].size();j++)
            {
                v=x[i-1][j];
                if(v>=(1<<n)) break;
                for(int k=0;k<n;k++)
                {
                    if((1<<k)&v) continue;
                    dp[i][(1<<k)|v]=max(dp[i][(1<<k)|v],dp[i-1][v]+a[i][k]);
                }
            }
        }
        printf("Case %d: %d\n",cas++,dp[n][(1<<n)-1]);
    }
    return 0;
}