CF3B-枚举

CF3B

题目描述

更正：第二行输出方案

题解

显然直接01背包肯定会T，但观察题目，重量只有两种情况，因此我们对重量为1和为2的两类物品进行从大到小的排序，然后枚举选取重量为1的数量，更新答案。$O(nlogn)$
注意枚举的边界问题

代码

#include<bits/stdc++.h>
#define M 500009
using namespace std;
int read(){
	int f=1,re=0;
	char ch;
	for(ch=getchar();!isdigit(ch)&&ch!='-';ch=getchar());
	if(ch=='-'){f=-1,ch=getchar();}
	for(;isdigit(ch);ch=getchar()) re=(re<<3)+(re<<1)+ch-'0';
	return re*f;
}
struct zb{
	int x,id;
}a[M],b[M];
bool cmp(const zb &a,const zb &b){return a.x>b.x;}
int n,v,cnt1,cnt2,num1,num2,sum1[M],sum2[M],ans;
int main(){
	n=read(),v=read();
	for(int i=1;i<=n;i++){
		int x=read(),y=read();
		if(x==1) a[++cnt1].x=y,a[cnt1].id=i;
		else b[++cnt2].x=y,b[cnt2].id=i;
	}sort(a+1,a+cnt1+1,cmp),sort(b+1,b+cnt2+1,cmp);
	for(int i=1;i<=cnt1;i++) sum1[i]=sum1[i-1]+a[i].x;
	for(int i=1;i<=cnt2;i++) sum2[i]=sum2[i-1]+b[i].x;
	for(int i=0;i<=min(cnt1,v);i++){
		int w=(v-i)/2;
		if(w>cnt2) w=cnt2;
		if(sum1[i]+sum2[w]>ans){
			ans=sum1[i]+sum2[w];
			num1=i,num2=w;
		}
	}printf("%d\n",ans);
	for(int i=1;i<=num1;i++) printf("%d ",a[i].id);
	for(int i=1;i<=num2;i++) printf("%d ",b[i].id); 
	return 0;
}