【CQBZOJ 2368】[线段树]球星

题目描述

线段树水题专项赛：Portal
给出球星们的能力值、年份、名字，有很多个查询，每个查询给出一个年份的范围，求出这个范围里能力值从高到低排列的前11名球员，如果能力值相同则按年份从低到高排，如果年份仍然相同，则按名字的字典序排。如果不足11个球员，就用XXX代替输出凑够11行。

题目分析

刚开始想线段树套Splay去了，考试要结束时才发现根本不用Splay啊，他只查询前11个，于是往每个线段树节点里塞一个优先队列，维护一下就行了。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<set>
#include<vector>
#include<algorithm>
#include<cstdlib>
#include<queue>
using namespace std;

#define MAXN 50000
#define MAXM
#define MAXLOG 25
#define INF 0x3f3f3f3f
typedef long long int LL;

template<class T>
void Read(T &x){
    x=0;char c=getchar();bool flag=0;
    while(c<'0'||'9'<c){if(c=='-')flag=1;c=getchar();}
    while('0'<=c&&c<='9'){x=x*10+c-'0';c=getchar();}
    if(flag)x=-x;
}

struct star{
    int year,ability;
    char name[16];
    bool operator == (const star &x)const{
        return ability==x.ability&&year==x.year&&strcmp(name,x.name)==0;
    }
    bool operator < (const star &x)const{
        if(ability!=x.ability)return ability>x.ability;
        else if(year!=x.year)return year<x.year;
        else return strcmp(name,x.name)<=0;
    }
};

bool cmp(const star &x,const star &y){return x.year<y.year;}


star A[MAXN+10];
int n,m;
int l,r;
priority_queue<star>ans;

priority_queue<star>val[MAXN*3+100];
star tmp[15];
int L[MAXN*3],R[MAXN*3];

void Build(int l,int r,int x){
    L[x]=l,R[x]=r;

    if(l==r)return;

    int mid=(l+r)>>1;
    Build(l,mid,x<<1);
    Build(mid+1,r,x<<1|1);
}

void Add(star &w,int x){
    val[x].push(w);
    while(val[x].size()>11)val[x].pop();
    if(L[x]==R[x])return;

    int mid=(L[x]+R[x])>>1;
    if(w.year<=mid)Add(w,x<<1);
    else Add(w,x<<1|1);
}

void Query(int x){
    if(l<=L[x]&&R[x]<=r){
        int len=0;
        while(!val[x].empty()){
            tmp[++len]=val[x].top();
            val[x].pop();
        }
        for(int i=1;i<=len;++i){
            ans.push(tmp[i]);
            if(ans.size()>11)ans.pop();
            val[x].push(tmp[i]);
        }

        return;
    }

    int mid=(L[x]+R[x])>>1;
    if(l<=mid)Query(x<<1);
    if(r>mid)Query(x<<1|1);
}

int p[MAXN+10],cnt;
int find(int w){
    int l=0,r=n,mid;
    int ans=0;
    while(l<=r){
        mid=(l+r)>>1;
        if(p[mid]<w){
            ans=mid;
            l=mid+1;
        }
        else r=mid-1;
    }
    return ans;
}

int main(){
    //freopen("star.in","r",stdin);
    //freopen("star.out","w",stdout);

    Read(n);

    for(int i=1;i<=n;++i)
        scanf("%d%s%d",&A[i].year,A[i].name,&A[i].ability);
    sort(A+1,A+n+1,cmp);

    cnt=1;
    for(int i=1;i<=n;++i){
        p[i]=A[i].year;
        if(A[i].year!=A[i+1].year)A[i].year=cnt++;
        else A[i].year=cnt;
    }

    --cnt;
    Build(1,cnt,1);

    for(int i=1;i<=n;++i)
        Add(A[i],1);

    Read(m);
    for(int i=1;i<=m;++i){
        Read(l),Read(r);
        while(!ans.empty())ans.pop();

        l=A[find(l)+1].year;
        r=A[find(r+1)].year;

        Query(1);

        int now=0;
        while(!ans.empty()){
            tmp[++now]=ans.top();
            ans.pop();
        }

        sort(tmp+1,tmp+now+1);

        for(int i=1;i<=now;++i)
            printf("%s\n",tmp[i].name);
        for(int i=now+1;i<=11;++i)
            puts("XXX");

        putchar('\n');
    }
}