P4291 [HAOI2008]排名系统

传送门

怎么主要写的都是平衡树,这种查询排名,查询第 $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;
}

 

posted @ 2019-09-03 11:08  LLTYYC  阅读(189)  评论(0编辑  收藏  举报