zoukankan      html  css  js  c++  java
  • BZOJ 2588 主席树

    思路:
    主席树
    做完BZOJ 3123 觉得这是道水啊……
    然后狂RE 狂MLE
    要来数据
    忘把deep[1]设成1了………………………….
    啊wocccccccccccccccc

    //By SiriusRen
    #include <cstdio>
    #include <cstring>
    #include <algorithm>
    using namespace std;
    #define N 200050
    #define M 5000000
    typedef long long ll;
    int n,m,xx,yy,zz,a[N],inf,cnt,fa[N/2][21],tree[M],lson[M],rson[M];
    int first[N],next[N],v[N],root[N],tot,ans,jy,deep[N];
    void add(int x,int y){v[tot]=y,next[tot]=first[x],first[x]=tot++;}
    void insert(int x,int &y,int l,int r,int wei){
        if(!y)y=++cnt;tree[y]=tree[x]+1;if(l==r)return;
        int mid=((ll)l+r)/2;
        if(mid<wei)lson[y]=lson[x],insert(rson[x],rson[y],mid+1,r,wei);
        else rson[y]=rson[x],insert(lson[x],lson[y],l,mid,wei);
    }
    void dfs(int x){
        for(int i=1;i<=20;i++)fa[x][i]=fa[fa[x][i-1]][i-1];
        insert(root[fa[x][0]],root[x],1,inf,a[x]);
        for(int i=first[x];~i;i=next[i])if(v[i]!=fa[x][0]){
            deep[v[i]]=deep[x]+1;
            fa[v[i]][0]=x,dfs(v[i]);
        }
    }
    int lca(int x,int y){
        if(deep[x]<deep[y])swap(x,y);
        for(int i=20;~i;i--)if(deep[fa[x][i]]>=deep[y])x=fa[x][i];
        if(x==y)return x;
        for(int i=20;~i;i--)if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
        return fa[x][0];
    }
    int query(int x,int y,int lca,int flca,int l,int r,int wei){
        if(l==r)return l;
        int mid=((ll)l+r)/2,temp=tree[lson[x]]+tree[lson[y]]-tree[lson[lca]]-tree[lson[flca]];
        if(temp<wei)return query(rson[x],rson[y],rson[lca],rson[flca],mid+1,r,wei-temp);
        else return query(lson[x],lson[y],lson[lca],lson[flca],l,mid,wei);
    }
    signed main(){
        memset(first,-1,sizeof(first)),scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)scanf("%d",&a[i]),inf=max(inf,a[i]);
        for(int i=1;i<n;i++)scanf("%d%d",&xx,&yy),add(xx,yy),add(yy,xx);
        deep[1]=1,dfs(1);
        for(int i=1;i<=m;i++){
            scanf("%d%d%d",&xx,&yy,&zz),xx^=ans;jy=lca(xx,yy);
            printf("%d",ans=query(root[xx],root[yy],root[jy],root[fa[jy][0]],1,inf,zz));
            if(i!=m)putchar('
    ');
        }
    }

    这里写图片描述

  • 相关阅读:
    java连接远程linux的redis
    Mac下Sublime Text 3安装配置
    矩阵覆盖
    Mac下配置Tomcat
    用 O(1) 时间检测整数 n 是否是 2 的幂次。
    快速编程之禅
    如何在centos 7.4 上安装 python 3.6
    大众点评实时监控系统CAT的那些坑
    如何在 centos 7.3 上安装 caffe 深度学习工具
    为什么中文编程项目失败率特别高?
  • 原文地址:https://www.cnblogs.com/SiriusRen/p/6532090.html
Copyright © 2011-2022 走看看