UOJ 52 元旦激光炮

http://uoj.ac/problem/52

题意:每次可以得到3个序列中

思路:每次分别取出三个序列的K/3长度的位置，取最小的那个，然后每次减掉它，总复杂度是Nlog3N

#include "kth.h"
#include<algorithm>
int C[];
int query_kth(int n_a, int n_b, int n_c, int k)
{   

    int Len=k;
    int la=,lb=,lc=;
    int ans=;
    while (Len>){
        int va=,vb=,vc=;
        int lena=std::min(Len/,n_a-la+);
        int lenb=std::min(Len/,n_b-lb+);
        int lenc=std::min(Len/,n_c-lc+);
        if (lena>=) va=get_a(la+lena-);
        if (lenb>=) vb=get_b(lb+lenb-);
        if (lenc>=) vc=get_c(lc+lenc-);
        if (va<=vb&&va<=vc){
           la+=lena;
           Len-=lena;
           ans=std::max(ans,va);
        }else
        if (vb<=va&&vb<=vc){
           lb+=lenb;
           Len-=lenb;
           ans=std::max(ans,vb);
        }else
        if (vc<=va&&vc<=vb){
           lc+=lenc;
           Len-=lenc;
           ans=std::max(ans,vc);
        }
    }
    int cnt=;
    int lena=std::min(Len,n_a-la+);
    int lenb=std::min(Len,n_b-lb+);
    int lenc=std::min(Len,n_c-lc+);
    for (int i=;i<=lena;i++)
        C[++cnt]=get_a(la+i-);
    for (int i=;i<=lenb;i++)
        C[++cnt]=get_b(lb+i-);
    for (int i=;i<=lenc;i++)
        C[++cnt]=get_c(lc+i-);
    std::sort(C+,C++cnt);
    for (int i=;i<=Len;i++)
        ans=std::max(ans,C[i]);

    return ans;

}