版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/jerry99s/article/details/81983697
题目:
http://acm.hdu.edu.cn/showproblem.php?pid=6435
题意:
有n个主武器;
有m个副武器;
每个武器有一个主属性值S,k个副属性值X[i,j](1<=i<=n,1<=j<=k);
选一种主武器i+一种副武器j;
使得Si + Sj +∑ | X[i,k] - X[j,k] |最大;
求最大值。
分析:
首先,|Xi-Xj| = max(Xi-Xj,Xj-Xi);
所以可以枚举第i种属性在前还是在后;
由于k<=5很小,可以状压;
状态s中第i位为1,表示第i种属性在前;
一个数组A[s]表示主武器中,s状态的最大值;
一个数组B[s]表示副武器中,s状态的最大值;
那么答案只需要枚举所有状态s,取max(A[s]+B[((1<< k)-1) xor s]);
复杂度O( n*k*(2^k) )
代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll inf=1e18;
int n,m,k,top;
ll x[10],A[100],B[100];
int main()
{
int T;
ll sum,ini;
cin>>T;
while(T--)
{
scanf("%d%d%d",&n,&m,&k);
top=(1<<k)-1;
for(int s=0;s<=top;s++)
A[s]=B[s]=-inf;
for(int i=1;i<=n;i++)
{
scanf("%I64d",&ini);
for(int j=1;j<=k;j++)
scanf("%I64d",&x[j]);
for(int s=0;s<=top;s++)
{
sum=ini;
for(int j=1;j<=k;j++)
{
if((s>>(j-1))&1) sum+=x[j];
else sum-=x[j];
}
A[s]=max(A[s],sum);
}
}
for(int i=1;i<=m;i++)
{
scanf("%I64d",&ini);
for(int j=1;j<=k;j++)
scanf("%I64d",&x[j]);
for(int s=0;s<=top;s++)
{
sum=ini;
for(int j=1;j<=k;j++)
{
if((s>>(j-1))&1) sum+=x[j];
else sum-=x[j];
}
B[s]=max(B[s],sum);
}
}
ll ans=0;
for(int s=0;s<=top;s++)
ans=max(ans,A[s]+B[top xor s]);
printf("%I64d\n",ans);
}
return 0;
}