E Camel Train
题意:n个骆驼排队,如果第i个骆驼排在第Ki位之前,就会产生Li的快乐值,否则会产生Ri的快乐值。求最大快乐值。
思路:
不妨设Li>Ri的骆驼排在Li<RI的骆驼的左边(其他情况不会更优),那么久可以把两类骆驼分开考虑。
对于Li>RI的骆驼,至少可以产生Ri的快乐值,那么我们要考虑的是哪些骆驼能如愿站在前KI位,多产生LI-Ri的快乐值。贪心,按照Ki枚举,加入队列,如果队列中超过i个骆驼,那么弹出LI-RI最小的骆驼即可。RI>Li的同理。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<vector>
#define LL long long
using namespace std;
const int N=2e5+10;
struct NODE{
int k,l,r;
bool operator <(const NODE &other)
const { return k<other.k; }
}node[N];
priority_queue<int,vector<int>,greater<int> >que;//小根堆,注意空格!!!!
int main()
{
int T,n;
scanf("%d",&T);
while(T)
{
T--;
LL sum=0;
LL now=0;
int cnt=0;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d%d%d",&node[i].k,&node[i].l,&node[i].r);
now+=max(node[i].l,node[i].r);
}
sort(node+1,node+n+1);
int j=1;
for(int i=1;i<=n;i++)
{
while(j<=n&&node[j].k<=i)
{
if(node[j].l>=node[j].r) //注意等于的情况不要漏掉
{
sum+=node[j].r;
que.push(node[j].l-node[j].r);
}
j++;
}
while(que.size()>i) que.pop();
}
while(que.size())
{
sum+=que.top();
cnt++;
que.pop();
}
j=n;
for(int i=n;i>0;i--)
{
while(j&&node[j].k>=i)
{
if(node[j].r>node[j].l)
{
sum+=node[j].l;
que.push(node[j].r-node[j].l);
}
j--;
}
while(que.size()>n-i) que.pop();
}
while(que.size())
{
sum+=que.top();
cnt++;
que.pop();
}
cout<<sum<<endl;
}
return 0;
}