3B - Lorry 预处理 + 暴力枚举

原题链接

注意细节，不要越界，还有题目对英文不好的我，真是不友好，呜呜。

题意：给你两种类型的卡车，一种体积是1，另一种是2，每辆车都是自己的载重量，现在给你一个体积，问你这个体积下，最大的载重量是多少？并给出方案车辆编号。

思路：排序后预处理前缀和， \(O(n)\)枚举即可。

代码如下

struct node{
	ll w;
	int id;
}a[N], b[N];
ll s1[N], s2[N];
int idx1, idx2;
int n, v;

bool comp(node a, node b)
{
	return a.w > b.w;
}

int main()
{
	IOS;
	cin >> n >> v;
	idx1 = idx2 = 1;
	for(int i = 1 ; i <= n ; i ++)
	{
		int t, p;
		cin >> t >> p;
		if(t == 1) a[idx1 ++] = {p, i};
		else b[idx2 ++] = {p, i};
	}
	sort(a + 1, a + idx1 + 1, comp);
	sort(b + 1, b + idx2 + 1, comp);
	for(int i = 1 ; i < idx1 ; i ++) s1[i] = s1[i - 1] + a[i].w;
	for(int i = 1 ; i < idx2 ; i ++) s2[i] = s2[i - 1] + b[i].w;
	
	ll maxd = 0;
	int l = 0, r = 0;
	for(int i = 0 ; i < idx2 && i * 2 <= v ; i ++)
	{
		ll sum = s2[i] + s1[min(v - i * 2, idx1 - 1)];
		if(maxd < sum)
		{
			maxd = sum;
			l = min(v - i * 2, idx1 - 1);
			r = i;
		}
	}
	
	cout << maxd << endl;
	for(int i = 1 ; i <= l ; i ++) cout << a[i].id << " ";
	for(int i = 1 ; i <= r ; i ++) cout << b[i].id << " ";
	cout << endl;
	
	return 0;
}