HDU 5691 Sitting in Line(状压DP+思维建图)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5691

#include<bits/stdc++.h>
using namespace std;

#define debug puts("YES");
#define rep(x,y,z) for(int (x)=(y);(x)<(z);(x)++)
#define read(x,y) scanf("%d%d",&x,&y)
#define ll long long

#define lrt int l,int r,int rt
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define root l,r,rt
const int  maxn =1e6+5;
const int mod=1e9+7;
const int INF=-1600000006;
ll powmod(ll x,ll y){ll t; for(t=1;y;y>>=1,x=x*x%mod) if(y&1) t=t*x%mod; return t;}
ll gcd(ll x,ll y){return y?gcd(y,x%y):x;}
/*
题目大意：给定一系列数，有正有负，
有些数位置固定，要求求出一个序列，其权值函数：
任意相邻两位乘积和最大。

要有思维建图的思想，假设每个数是一个孤立的点，
现在要求找到一个欧拉通路，
并使得其权值最大。

明显经典的状压DP，开二维，i表示状态量，j表示末尾节点，
这样状态就可以扩展了，枚举i中所有的末尾节点，再枚举扩展节点，
找准判定条件这题就简单了。

根据题目要求，那么枚举的状态i的1的个数要等于j对应的位置数（如果指定了的话）。
最后枚举下最终状态的末尾节点DP值即可。

*/
int getcnt(int x){int ret=0;while(x) x-=x&(-x),ret++;return ret;}
int num[20],s[20],n;
ll dp[1<<17][17];
int cnt[1<<16];

int main()
{
    for(int i=0;i<(1<<16);i++) cnt[i]=getcnt(i);
    int t;scanf("%d",&t);
    for(int ca=1;ca<=t;ca++)
    {
        scanf("%d",&n);  for(int i=0;i<n;i++) scanf("%d%d",&num[i],&s[i]);
        for(int i=0;i<n;i++) for(int j=0;j<(1<<n);j++) dp[j][i]=INF;
        for(int i=0;i<n;i++)  if(s[i]==-1||s[i]==0) dp[1<<i][i]=0;
        for(int i=0;i<(1<<n);i++)
        {
            for(int j=0;j<n;j++)///找好当前位置
            {
                if(i&(1<<j)==0) continue;
                for(int k=0;k<n;k++)
                {
                    if(j==k) continue;
                    if(dp[i][j]==INF) continue;///当前状态不可扩展
                    if(s[k]==-1||cnt[i]==s[k])///只有当前位置可扩展或者扩展的位置数刚好是标志值才可以
                    {
                        if(i&(1<<k)) continue;///如果当前状态已经包含则没有必要扩展
                        if(dp[i|(1<<k)][k]<dp[i][j]+num[j]*num[k]) dp[i|(1<<k)][k]=dp[i][j]+num[j]*num[k];
                    }
                }
            }
        }
        ll ans=INF;  for(int i=0;i<n;i++) ans=max(ans,dp[(1<<n)-1][i]);
        printf("Case #%d:\n%lld\n",ca,ans);
    }
    return 0;
}