三元组
来自 GJX 大佬的题目
Description
给出
n
个三元组(
每一个三元组最多只能选择一个数(即
给出三个要求:
1、选择
xi
的三元组恰有
X
个。
2、选择
3、选择
问选出的数的和最大为多少。
Data Constraint
Solution
考虑若
X
=
因而对于三元组,可以先强制选择所有的
设新的二元组为(
将
n
个二元组按照
考虑枚举最后选择
vi
的位置为
i
,则
使用同样的方法,将
1
~
对于维护一些数中的前
k
大的和,看到数据范围
Code
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,j,l) for(int i=j;i<=l;i++)
#define fd(i,j,l) for(int i=j;i>=l;i--)
using namespace std;
typedef long long ll;
const ll N=55e4,M=2*N;
struct note{
ll y,z;
}t[N];
int X,Y,Z;
ll x[N],t1[3*N],t2[3*N];
int n,m,j,k,l,i,o,p;
int max(int a,int b)
{if(a>b)return a;else return b;}
int read()
{
int o=0; char ch=' ';
for(;ch<'0'||ch>'9';)ch=getchar();
for(;ch>='0'&&ch<='9';ch=getchar())o=o*10+ch-48;
return o;
}
bool kmp(note a,note b)
{return a.y!=b.y ? a.y>b.y : a.z<b.z ;}
int main()
{
cin>>X>>Y>>Z; n=X+Y+Z;
fo(i,1,n)x[i]=read(),t[i].y=read(),t[i].z=read();
ll ans=0,da=0;
fo(i,1,n)da+=x[i],t[i].y-=x[i],t[i].z-=x[i];
sort(t+1,t+n+1,kmp);
ll dq=0,op=0,u=0,zd=M+N-10,v=0;
fo(i,1,Y)t1[t[i].z-t[i].y+M]++,dq+=t[i].y;
fo(i,Y+1,n)t2[t[i].z+M]++;
for(;!t1[zd];)zd--;
v=t1[zd]; t1[zd]=0;
int rr=M+N-10,sx=Z;
while(t2[rr]<sx)op+=t2[rr]*(rr-M),sx-=t2[rr],rr--;
u=rr; op+=sx*(rr-M); t2[rr]=0; ans=op+dq;
fo(i,Y+1,Y+Z){
dq+=t[i].y; ll tt=t[i].z-t[i].y+M;
if(tt>zd)dq+=tt-M;
else if(tt==zd)dq+=tt-M;
else {
t1[tt]++;
dq+=zd-M; v--;
if(!v){
for(;!t1[zd];)zd--;
v=t1[zd]; t1[zd]=0;
}
}
tt=t[i].z+M;
if(tt>u)op-=tt-M,t2[tt]--;
else {
sx--,op-=u-M;
if(!sx){
for(;!t2[u];)u++;
sx=t2[u]; t2[u]=0;
}
}
ans=max(ans,dq+op);
}
printf("%lld",ans+da);
}