hdu5691 Sitting in Line（状压dp）

 1 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<cmath>
 using namespace std;
 #define max(a,b) a>b?a:b
 #define LL long long
 #define inf 0x3f3f3f3f
 int a[], b[], num[<<];
 LL dp[][<<]; // dp[x][y]  x表示数组最后一个元素，y表示当前状态
 int n;
 int getnum(int x){
     int ret=;
     while(x){
         ret++;
         x=x&(x-);
     }
     return ret;
 }
 void init(){
     for(int i=;i<(<<);++i)
     num[i]=getnum(i);
 }
 int main()
 {
     int t, i, j, k, cas=;
     init();
     scanf("%d",&t);
     while(t--){
         scanf("%d",&n);
         for(i=;i<n;++i)
             scanf("%d%d",&a[i],&b[i]);
         for(i=;i<n;++i){
             for(j=;j<(<<n);++j)
                 dp[i][j]=-inf;
         }
         for(i=;i<n;++i){
             if(b[i]==-||b[i]==)
             dp[i][<<i]=;
         }
         for(i=;i<(<<n);++i){
             for(j=;j<n;++j){
                 if(i&(<<j)==) continue;
                 if(dp[j][i]==-inf) continue;
                 for(k=;k<n;++k){
                     if(j==k) continue;
                     if(i&(<<k)) continue;
                     if(b[k]==-||b[k]==num[i]){
                         dp[k][i|(<<k)]=max(dp[k][i|(<<k)],
                         dp[j][i]+a[j]*a[k]);
                     }
                 }
             }
         }
         LL ans=-inf;
         for(i=;i<n;++i){
             ans=max(ans,dp[i][(<<n)-]);
         }
         printf("Case #%d:\n",cas++);
         printf("%lld\n",ans);
     }
     return ;
 }