JZOJ 5432 三元组

三元组

来自$GJX$大佬的题目

Description

给出$n$个三元组($x_i$,$y_i$,$z_i$)及三个整数$X$,$Y$,$Z$满足$X$+$Y$+$Z$=$n$。
每一个三元组最多只能选择一个数(即$x_i$,$y_i$,$z_i$中的一个)
给出三个要求:
1、选择$x_i$的三元组恰有$X$个。
2、选择$y_i$的三元组恰有$Y$个。
3、选择$z_i$的三元组恰有$Z$个。
问选出的数的和最大为多少。

Data Constraint

$1$<=$n$<=$5*10^5$，$0$<=$x_i$,$y_i$,$z_i$<=$5*10^5$

Solution

考虑若$X$=$0$时怎么做，先强制选择所有的$y_i$，然后按照$z_i$-$y_i$排序，选择最大的$Z$个加入答案中。
因而对于三元组，可以先强制选择所有的$x_i$，然后从$n$个新的二元组($y_i$-$x_i$,$z_i$-$x_i$)中选择$Y$个$y_i-x_i$和$Z$个$z_i-x_i$使价值最大。(有$X$个二元组不选择)

设新的二元组为($v_i$,$u_i$)。
将$n$个二元组按照$v_i$为第一关键字从大到小排序。
考虑枚举最后选择$v_i$的位置为$i$，则$1$~$i$的位置一定都被选择了，如果不是，则最后选择$v_i$的位置一定在$i$前面，$significantly$。
使用同样的方法，将$1$~$i$的$v_i$全部都强制选择，然后从前$i$个二元组中选择$i$-$Y$个最大的$u_i$-$v_i$，从$i$+$1$~$n$个二元组中选择$Z$-($i$-$Y$)个最大的$u_i$，两块价值加起来与待定答案取$max$即可。

对于维护一些数中的前$k$大的和，看到数据范围$0$<=$x_i$,$y_i$,$z_i$<=$5*10^5$，可以开两个桶维护区间$1$~$i$和$i$+$1$~$n$，维护两个指针分别指向第$k$大的位置，然后可以发现随着$i$的增加，指针只会往单方向移动，因此时间复杂度可以做到$O$( n <script type="math/tex" id="MathJax-Element-7693">n</script>)。

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);
}