HDU5961 Sitting in Line【状压DP】

题目链接：HDU5961 Sitting in Line

题意：给你n个数和他们的位置，如果位置为-1表示这个数可以取任意位置，选择一种放置方式，使得相邻两数乘积之和最大；

分析：N<=16，考虑状压DP；dp[i][j]表示取状态i中为1的数放在前面且这一位放j时的答案；当dp[i][j]成立时，可以从它向后更新dp[i|(1<<k)][k]，要注意更新时需要判断这一位的位置可以随意放或者它需要放的位置正好是当前处理的位置；

#include<bits/stdc++.h>
using namespace std;
const int INF=0x3f3f3f3f;
int dp[(1<<17)+1][17],a[17],p[17];
//dp[i][j] 取状态i中为1的数放在前面且这一位放j时的答案
int rua()
{
    int n;scanf("%d",&n);
    for(int i=0;i<n;i++) scanf("%d%d",&a[i],&p[i]);
    memset(dp,-INF,sizeof(dp));
    for(int i=0;i<n;i++) if(p[i]<=0) dp[(1<<i)][i]=0;
    for(int i=0;i<(1<<n);i++)
        for(int j=0;j<n;j++)
        {
            if(dp[i][j]==-INF) continue;
            for(int k=0;k<n;k++)
                if(((i&(1<<k))==0) && (__builtin_popcount(i)==p[k] || p[k]==-1))
                    dp[i|(1<<k)][k]=max(dp[i|(1<<k)][k],dp[i][j]+a[j]*a[k]);
        }
    int ans=-INF;
    for(int i=0;i<n;i++) ans=max(ans,dp[(1<<n)-1][i]);
    return ans;
}
int main()
{
    int t;scanf("%d",&t);
    for(int id=1;id<=t;id++) printf("Case #%d:\n%d\n",id,rua());
    return 0;
}