POJ 3900 The Robbery

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <iostream>
using namespace std;
long long maxa(long long a,long long b)
{
return a>b?a:b;
}

struct node
{
long long w;
long long v;
double vbw;
int id;
}Q[20];

int n,m;
long long ans;
long long sumv[20];
long long tot;

bool cmp(node a,node b)
{
return a.vbw>b.vbw;
}

void dfs(int pos,long long sum,int left)//分别是 搜到的当前位置   当前的总价值  当前还剩多少重量
{
ans=maxa(ans,sum);
if(pos>n)return;  //越界

if(sum + sumv[pos-1] <=ans)return;//如果剩下的全背上都不比当前最优解好。就剪掉

if(sum + left * Q[pos].vbw <=ans)return; //在性价比排序的基础上，如果剩下的重量全部背这种性价比最高的还不能比最优解好。就剪掉

for(int k=Q[pos].id;k>=0;k--)
{
if(left - k*Q[pos].w <0)continue;
dfs(pos+1,sum+k*Q[pos].v,left-k*Q[pos].w);
}
}

int main()
{
int T;
scanf("%d",&T);
while(T--)
{
ans=0;
scanf("%lld%lld",&n,&m);
tot=0;
for(int i=1;i<=n;i++)
{
scanf("%lld",&Q[i].w);
Q[i].id = i;
}
for(int i=1;i<=n;i++)
{
scanf("%lld",&Q[i].v);
tot+=i*Q[i].v;
}

for(int i=1;i<=n;i++)
Q[i].vbw=1.0 * Q[i].v/Q[i].w;

sort(Q+1,Q+n+1,cmp);
sumv
=0;

for(int i=n;i>=1;i--)
{
//printf("i = %d  Q[i].id = %d\n",i,Q[i].id);
sumv[i-1] = sumv[i] + Q[i].id*Q[i].v;   //求出后缀和  不包括自身
}
/*for(int i=0;i<=n;i++)
printf("%lld ",sumv[i]);*/

dfs(1,0,m);

printf("%lld\n",ans);
}
return 0;
}