【5691】Sitting in Line

5691
状压DP

//__builtin_popcount()计算一个 32 位无符号整数有多少个位为1
#include<bits/stdc++.h>
using namespace std;
const int inf=0x3f3f3f3f;
int a[20],p[20],dp[1<<20][20];
int main(){
	int t,n;
	scanf("%d",&t);
	for(int cas=1;cas<=t;cas++){
		scanf("%d",&n);
		for(int i=0;i<n;i++)
			scanf("%d%d",&a[i],&p[i]);
		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++)
			if(p[i]==0||p[i]==-1)
				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)
					for(int k=0;k<n;k++)
						if(((1<<k)&i)==0&&(p[k]==__builtin_popcount(i)||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]);
		printf("Case #%d:\n%d\n",cas,ans);
	}
	return 0;
}