学园偶像的盛会

2021-10-09 08:00:34 阅读：210 来源： 互联网

Description

\(n\) 个乐队，舞台构成一棵树。第 \(i\) 个乐队的名字叫 \(s_i\)，会从第 \(i\) 个节点向上的所有 \(y_i\) 个节点依次巡演。对于一个点，假设其上有 \(k\) 支队伍要在此巡演，那么她们会按名字的字典序大小依次表演。\(m\) 个询问，每次询问乐队 \(x\) 第 \(k\) 次表演之前是哪个乐队，输出名字，不存在输出 "none"。

Solution

对于一个乐队，在其初始位置打上加法标记，在结束位置的父亲打上减法标记。离线询问，丢到节点上。前面这部分预处理可以倍增优化。由于名字很短，直接使用自带的 strcmp 来快排，处理出排名。每个节点维护一棵主席树，答案实际上就是前驱。所以 dfs 一遍整棵树，处理每个点的询问，然后向上线段树合并即可。

\(O(n\log n)\)，题解写的是平衡树启发式合并，会多一个 \(\log\)，不太优秀。

#include<stdio.h>
#include<algorithm>
#include<vector>
#include<string.h>
using namespace std;

inline int read(){
    int x=0,flag=1; char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')flag=0;c=getchar();}
    while(c>='0'&&c<='9'){x=(x<<1)+(x<<3)+c-48;c=getchar();}
    return flag? x:-x;
}

const int N=2e5+7;

char s[N][14];
int fa[N][17],a[N],head[N],n,m;

struct Node{
    int sz,ls,rs;
}t[N*20];

struct E{
    int next,to;
}e[N<<1];

struct Query{int val,pos;};
vector<int> C[N];
vector<Query> Q[N];

int ans[N],rk[N],Rk[N];

inline void add(int id,int to){
    static int cnt=0;
    e[++cnt]=(E){head[id],to};
    head[id]=cnt;
    e[++cnt]=(E){head[to],id};
    head[to]=cnt;
}

void dfs(int u){
    for(int i=1;i<=16;i++)
        fa[u][i]=fa[fa[u][i-1]][i-1];
    for(int i=head[u];i;i=e[i].next)
        if(e[i].to!=fa[u][0]) fa[e[i].to][0]=u,dfs(e[i].to);
}

inline void update(int id){
    t[id].sz=t[t[id].ls].sz+t[t[id].rs].sz;
}

int Pos;
void modify(int &id,int lf,int rf){
    static int cnt=0;
    if(!id) id=++cnt;
    if(lf==rf) t[id].sz^=1;
    else{
        int mid=(lf+rf)>>1;
        if(Pos<=mid) modify(t[id].ls,lf,mid);
        else modify(t[id].rs,mid+1,rf);
        update(id);
    }
}

int merge(int x,int y,int lf,int rf){
    if(!x||!y) return x+y;
    if(lf==rf) t[x].sz+=t[y].sz;
    else{
        int mid=(lf+rf)>>1;
        t[x].ls=merge(t[x].ls,t[y].ls,lf,mid);
        t[x].rs=merge(t[x].rs,t[y].rs,mid+1,rf);
        update(x);
    }
    return x;
}

int query(int id,int lf,int rf){
    if(!id||!t[id].sz) return 0;
    if(lf==rf) return Rk[lf];
    int mid=(lf+rf)>>1;
    if(Pos<=mid) return query(t[id].ls,lf,mid);
    int ret=query(t[id].rs,mid+1,rf);
    if(ret) return ret;
    return query(t[id].ls,lf,mid);
}

int rt[N];
void Dfs(int u){
    for(int i=head[u];i;i=e[i].next){
        const int v=e[i].to;
        if(v==fa[u][0]) continue;
        Dfs(v),rt[u]=merge(rt[u],rt[v],1,n);
    }
    for(int x:C[u])
        Pos=x,modify(rt[u],1,n);
    for(Query x:Q[u])
        Pos=x.val-1,ans[x.pos]=Pos? query(rt[u],1,n):0;
}

inline bool Cmp(int x,int y){
    return strcmp(s[x],s[y])==-1;
}

int main(){
    freopen("idol.in","r",stdin);
    freopen("idol.out","w",stdout);
    n=read(),m=read();
    for(int i=1;i<=n;i++)
        scanf("%s %d",s[i],&a[i]),Rk[i]=i;
    for(int i=2;i<=n;i++) add(read(),read());
    dfs(1),sort(Rk+1,Rk+1+n,Cmp);
    for(int i=1;i<=n;i++) rk[Rk[i]]=i;
    for(int i=1;i<=m;i++){
        int x=read(),y=read()-1,now=x;
        for(int j=16;~j;j--)
            if(y&(1<<j)) now=fa[now][j];
        if(!now||y+1>a[x]) continue;
        Q[now].push_back((Query){rk[x],i});
    }
    for(int i=1;i<=n;i++){
        C[i].push_back(rk[i]);
        int now=i;
        for(int j=16;~j;j--)
            if(a[i]&(1<<j)) now=fa[now][j];
        if(now) C[now].push_back(rk[i]);
    }
    Dfs(1);
    for(int i=1;i<=m;i++)
        printf("%s\n",ans[i]? s[ans[i]]:"none");
}

标签：lf,cnt,include,int,盛会,学园,rf,偶像,id
来源： https://www.cnblogs.com/wwlwQWQ/p/15383880.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

学园偶像的盛会

Description

Solution