两种船,大小分别为1、2。现有一辆容量为v的卡车,给出n艘船类型,以及相应的价值,求最大装载价值以及所选的船只编号。
贪心策略:将大小为1、2的船分开来,按照价值从大到小排序。然后按价值由大到小,先装入大小为2的船直到装不下,再装大小为1的船。记录下此时大小为1、2的船个数n1,n2,以及总价值。然后把已装入的价值最小的大小为2的船拿出,换成两艘大小为1的价值高的船,看价值是否增加,若增加更新总价值,否则答案就是之前的总价值(因为不可能再存在两艘大小为1的船总价值超过已装入的大小为2的船了)
对于输出方案,在输入时记录下船的编号即可。
#include<bits/stdc++.h>
using namespace std;
struct Boat
{
int p,id;
};
Boat b1[100005],b2[100005];
int b[100005];
bool cmp(Boat x,Boat y)
{
if(x.p==y.p) return x.id<y.id;
return x.p>y.p;
}
int main()
{
int i,j,t,n,v,t1,t2;
cin>>n>>v;
t1=t2=1;
for(i=1;i<=n;++i)
{
cin>>t;
if(t==1) {scanf("%d",&b1[t1].p);b1[t1++].id=i;}
else {scanf("%d",&b2[t2].p);b2[t2++].id=i;}
}
sort(b1+1,b1+t1,cmp);
sort(b2+1,b2+t2,cmp);
int ans=0,n2,n1;
for(n2=1;n2<t2&&v>=2;++n2) {v-=2;ans+=b2[n2].p;}
for(n1=1;n1<t1&&v>=1;++n1) {v-=1;ans+=b1[n1].p;}
int temp=ans;
for(i=n2-1;i>=1;--i)
{
temp-=b2[i].p;
v+=2;
for(j=n1;j<=t1&&v>=1;++j) {v--;temp+=b1[j].p;}
if(temp>ans)
{
--n2;
ans=temp;
n1=j;
}
else break;
}
for(i=1;i<n2;++i) b[i]=b2[i].id;
for(i=1;i<n1;++i) b[i+n2-1]=b1[i].id;
printf("%d\n",ans);
for(i=1;i<=n1+n2-2;++i) printf("%d ",b[i]);
return 0;
}