分析:
数字放在a[i]中,标记放在b[i]中
原本以为是简单的贪心,结果发现局部最优并不能推出整体最优。后来仔细一看发现是一个状压dp。
dp[i][j] 表示状态为i以a[i]元素结尾,状态为j的最大和。( 状态的意思是当前已经选择了哪些数。 )
那么答案就是 dp[i][(1<<n)-1] { 0<=i<n } 中的最大值
代码:
#include <stdio.h>
#include <string.h>
#define INF 1700000000
int dp[17][(1<<16)-1];//dp[j][k] 表示第j个数结尾状态为k的最大值
int a[17];
int b[17];
int n;
int num(int x) {
int ans=0;
while(x) {
if(x%2==1)
ans++;
x>>=1;
}
return ans;
}
int MAX(int a,int b)
{
return a>b?a:b;
}
int main()
{
int t,i,j,k,Case = 0;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
for(i = 0 ; i < n ; i ++)
scanf("%d%d",&a[i],&b[i]);
for(j = 0 ; j < n ; j ++)
for(k = 0 ; k < (1<<n) ; k ++)
dp[j][k] = -INF;
for(i = 0 ; i < n ; i ++)
{
if(b[i] == -1 || b[i] == 0)
{
dp[i][(1<<i)] = 0;
}
}
for(i = 0 ; i <(1<<n) ; i ++)
{
for(j = 0 ; j < n ; j ++)
{
if(dp[j][i] != -INF)
{
for(k = 0 ; k < n ; k ++)
{
if((i&(1<<k)) == 0 &&(b[k] == -1 || b[k] == num(i)))
{
dp[k][i|(1<<k)] = MAX(dp[k][i|(1<<k)],dp[j][i] + a[j]*a[k]);
}
}
}
}
}
int ans = -INF;
for(i= 0 ; i < n ; i ++)
ans = MAX(ans,dp[i][(1<<n)-1]);
printf("Case #%d:\n%d\n",++Case,ans);
}
return 0;
}