思路:定义dp【i】【j】:i状态下,以a[j]结尾的最大排列 ,i为0到1<<n。(用二进制的方式表示:如果该位上的数选择了,则对应位置位1,否则为0).状态转移:dp[i|(1<<k)][k]=max(dp[i|(1<<k)][k],dp[i][j]+a[i]*a[j])(a[k]排在a[j后,"|"相当于取并集,就是把a[k]加入状态i)。然后就要用到__builtin_popcount(),返回二进制数中“1”有多少个。
注意:
1、还有就是不知道为什么直接memset(dp,-inf,sizeof(dp))会TLE。。看来以后遇到无穷小的情况还是老老实实用for初始化。
2、下面的dp!=inf写成dp==0也会出错,原因主要 是卡了精度。。可是为什么会卡精度呢??
3、这里我们初始化第一个p[i]==0是因为位置0之前已经没有数可以初始化它了。
4、初始化p[i]==-1而不是其他是因为,如果初始化其他,当全部数位都不确定时写出来的程序又要复杂得多,直接初始化p[i]==-1就不用在意这些问题。
参考:http://blog.csdn.net/queuelovestack/article/details/51471639
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define inf 0x3f3f3f3f
#define debug cout<<"debug"<<endl;
using namespace std;
const int maxn=1e4+50;
int n,t,a[50],p[50];
long long ans,dp[1<<20][50];
int main(){
while(scanf("%d",&t)!=EOF){
for(int cas=1;cas<=t;cas++){
scanf("%d",&n);
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",&a[i],&p[i]);
if(p[i]==-1||!p[i]) 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)&&(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]);
}
}
}
}
ans=-inf;
for(int j=0;j<n;j++){
ans=max(ans,dp[(1<<n)-1][j]);
}
printf("Case #%d:\n%lld\n",cas,ans);
}
}
return 0;
}
总结:对于这种涉及到给出一串数(或叫做位置?),判断某一位是否选择可以用二进制枚举得方式,这样就要对位运算熟悉一点才行,如&代表求交集,|代表求并集等等。