3B - Lorry
作者:互联网
原题链接https://codeforces.com/problemset/problem/3/B
注意细节,不要越界,还有题目对英文不好的我,真是不友好,呜呜。
题意:给你两种类型的卡车,一种体积是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;
}
标签:int,s2,ll,++,idx2,3B,idx1,Lorry 来源: https://www.cnblogs.com/luoyicong/p/14612999.html