P4291 [HAOI2008]排名系统

Posted 2022-09-27 lltyyc

传送门

怎么主要写的都是平衡树，这种查询排名，查询第 $K$ 大的操作直接权值线段树就行了

把读入的数据离散化一波，然后开个 $map$ 维护每个人最后一次插入时在线段树上的位置，直接线段树维护就完事了

查询排名就询问大于它的节点数量，查询第 $K$ 大直接线段树上二分

就是数据格式比较恶心，细节有点多

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<map>
using namespace std;
typedef long long ll;
inline int read()

    int x=0,f=1; char ch=getchar();
    while(ch<‘0‘||ch>‘9‘)  if(ch==‘-‘) f=-1; ch=getchar(); 
    while(ch>=‘0‘&&ch<=‘9‘)  x=(x<<1)+(x<<3)+(ch^48); ch=getchar(); 
    return x*f;

const int N=5e5+7;
int n;
struct dat
    int scr,tim; string nam; char opt;
    inline bool operator < (const dat &tmp) const 
        return scr!=tmp.scr ? scr<tmp.scr : tim>tmp.tim;
    
c[N],d[N];
map <string,int> pre;
int t[N<<2],pos,v,res;
void add(int o,int l,int r)

    t[o]+=v; if(l==r) return;
    int mid=l+r>>1;
    if(pos<=mid) add(o<<1,l,mid);
    else add(o<<1|1,mid+1,r);

void query_rnk(int o,int l,int r)

    if(l==r) return;
    int mid=l+r>>1;
    if(pos<=mid) query_rnk(o<<1,l,mid),res+=t[o<<1|1];
    else query_rnk(o<<1|1,mid+1,r);

void query(int o,int l,int r)

    if(l==r)  res=l; return; 
    int mid=l+r>>1;
    if(v>t[o<<1|1]) v-=t[o<<1|1],query(o<<1,l,mid);
    else query(o<<1|1,mid+1,r);

inline int Val(string s)

    int m=s.size(),res=0;
    for(int i=0;i<m;i++) res=res*10+s[i]-‘0‘;
    return res;

int main()

    n=read();
    for(int i=1;i<=n;i++)
    
        cin>>d[i].opt>>d[i].nam;
        if(d[i].opt==‘+‘) d[i].scr=read();
        d[i].tim=i; c[i]=d[i];
    
    sort(d+1,d+n+1); int tot=0;
    for(int i=1;i<=n;i++)
    
        if(c[i].opt==‘+‘)
        
            pos=pre[c[i].nam];
            if(pos) v=-1,add(1,1,n);
            else tot++;
            pos=lower_bound(d+1,d+n+1,c[i])-d;
            v=1; add(1,1,n); pre[c[i].nam]=pos;
            continue;
        
        if(c[i].nam[0]>‘9‘||c[i].nam[0]<‘0‘)
        
            res=0; pos=pre[c[i].nam]; query_rnk(1,1,n);
            printf("%d\n",res+1); continue;
        
        for(int k=1,j=Val(c[i].nam); k<=10&&j<=tot; k++,j++)
        
            v=j; res=0; query(1,1,n);
            cout<<d[res].nam; printf(" ");
        
        printf("\n");
    
    return 0;