hdu 5691(状压DP) Sitting in Line

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

状态DP，dp[i][j]，i 表示的是一种状态，这个状态指的是当前这个数取或不取，j表示的是以第j个数结尾，

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

typedef long long ll;
ll dp[1<<17][20];

int num[20],dis[20],vis[20];
ll max(ll x,ll y) {return x>y?x:y;}
const ll INF=1e18;

int main()
{
    int t,n,pos=0;
    scanf("%d",&t);
    while (t--)
    {
        scanf("%d",&n);
        memset(vis,-1,sizeof(vis));
        for(int i=0;i<(1<<n);i++)
            for(int j=0;j<n;j++)
                dp[i][j] = -INF;
        for (int i=0 ; i<n ; i++) {
            scanf("%d%d",&num[i],&dis[i]);
            if (dis[i]!=-1)
                vis[dis[i]]=i;
        }
        if (vis[0]!=-1) dp[(1<<vis[0])][vis[0]]=0;// 0这个位子被占用了
        else{
            for (int i=0 ; i<n ; i++) // 没有特定位置
                if (dis[i]==-1) dp[(1<<i)][i]=0; //
        }
        int cas=(1<<n)-1;
        for (int i=1 ; i<=cas ; i++){
            int ans=0;
            for (int j=0 ; j<n ; j++)
                if (i&(1<<j))
                   ans++;
            if (vis[ans]!=-1)
            {
                ans=vis[ans];
                for (int j=0 ; j<n ; j++)
                    if ((i&(1<<j))&&j!=ans)
                        dp[i|(1<<ans)][ans]=max(dp[i|(1<<ans)][ans],dp[i][j]+num[j]*num[ans]);
            }
            else
            {
               for (int j=0 ; j<n ; j++){
                   if (i&(1<<j)){
                      for (int k=0 ; k<n ; k++){
                          if (!(i&(1<<k))){
                              dp[i|(1<<k)][k]=max(dp[i|(1<<k)][k],dp[i][j]+num[k]*num[j]);
                          }
                      }
                   }
               }
            }
        }
        printf("Case #%d:
",++pos);
        ll maxn=-INF;
        for (int i=0 ; i<n ; i++)
            maxn=max(maxn,dp[cas][i]);
        printf("%I64d
",maxn);
    }
    return 0;
}