转载:http://blog.csdn.net/qian99/article/details/39138329
题意:给出n个物品,每个物品有两种属性Wi,Ti,有q组查询,每组查询要求在n个物品中选出一些,并使得两个属性的和为Mi,Si。
思路:刚开始看感觉是神题,后来仔细想了想,其实本质上就是个背包。最裸着写的话,那么就是dp[i][j][k]表示使用前i个物品,是否可以凑出第一个属性j,第二个属性k,要输出方案的话记录一下路径就可以了。一开始这么写了一发,加了一些乱七八糟的优化,还是会T。虽然这题时限还算宽,但这么写复杂度还是太高了。考虑到第一个属性最多只有50,那么可以用一个二进制数来表示是否能凑出第一个属性的情况,即:第i位为1表示可以凑出i。使用这种方法的好处是对于物品i可以直接算出第一种属性的组合情况,枚举一下新增的位,更新一下结果就行了。
代码:
#include<iostream>#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-8
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxn = 400 + 10;
const int M = 51;
short ans[200010][52];
ull f[200010];
int W[maxn],T[maxn];
map<ull,int>mp;
int main()
{
// freopen("in.txt","r",stdin);
// freopen("out.txt","w",stdout);
for(int i = 1;i <= M + 1;++i)
mp[1LL<<(i-1LL)] = i;
int t,n,q;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&q);
memset(ans,0,sizeof(ans));
memset(f,0,sizeof(f));
f[0] = 1;
ull v,x;
for(int i = 1;i <= n;++i)
{
scanf("%d%d",&W[i],&T[i]);
for(int j = 200000;j >= T[i];--j)
{
v = f[j]; //f[j]表示第二个属性为j时,能够凑出的第一个属性的集合,用一个二进制数表示,第i位为1表示可以凑出这个数
f[j] |= (f[j - T[i]]<<W[i]) & ((1LL<<M+1) - 1); //计算使用当前物品能够得到的新的集合,在集合f[j - T[i]]添加W[i]的物品,
//即原来能得到的每个值加上W[i],等价于将其左移W[i]位
for(ull k = v ^ f[j];k ; k &= k-1) //枚举新增加的集合
{
x = (k ^ (k - 1)) & k;
ans[j][mp[x] - 1] = i; //将新增的位置更新,记录是使用了哪个物品达到的这个状态
}
}
}
int m,s,p;
for(int i = 0;i < q;++i)
{
scanf("%d%d",&m,&s);
if(!ans[s][m])
puts("No solution!");
else
{
printf("%d",ans[s][m]);
p = ans[s][m];
m -= W[p];
s -= T[p];
while(m)
{
p = ans[s][m];
printf(" %d",p);
m -= W[p];
s -= T[p];
}
puts("");
}
}
}
return 0;
}