LightOJ 1011 - Marriage Ceremonies(状态压缩DP)

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

using namespace std;

int a[20][20];
int dp[20][70000];

int main()
{
int t, n;
scanf("%d", &t);
int tt = 1;
while(t--)
{
scanf("%d", &n);
for(int i=1; i<=n; i++)
{
for(int j=1; j<=n; j++)
{
scanf("%d", &a[i][j]);
}
}
int maxn = 1<<n;
for(int i=0; i<=n; i++)
for(int j=0; j<=maxn; j++)
dp[i][j] = 0;
for(int i=1; i<=n; i++)
{
for(int j=0; j<maxn; j++)
{
int cnt = 0;
for(int k=0; k<n; k++)//一步优化 就是当前i-1的状态下有不是i-1个组成的话 pass掉
if(j & (1<<k))
cnt++;
if (cnt != i-1)
continue;
for(int k=1; k<=n; k++)
{
int st = 1<<(k-1);
if((st&j)==0)
{
dp[i][st|j] = max(dp[i][st|j], dp[i-1][j]+a[i][k]);
}
}
}
}

printf("Case %d: %d\n", tt++, dp
[(1<<n)-1]);
}

return 0;
}