zoukankan      html  css  js  c++  java
  • 洛谷 P2633 Count on a tree[bzoj2588](倍增lca+主席树)

    传送门
    此题非常裸,就是代码太多了,2519,115行。
    没什么可说的,思路就是每个节点利用它的父亲建立主席树,查询的时候,sum[u]+sum[v]-sum[lca[u]]-sum[father[lca[u,v]]]就是左子树的大小,然后就没什么了。
    代码:(1000ms,还挺快的)

    #include<cstdio>
    #include<algorithm>
    #define ll long long
    using namespace std;
    inline int read(){
        int x=0;char ch=' ';
        while(ch<'0'||ch>'9')ch=getchar();
        while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
        return x;
    }
    struct edge{
        int to,next;
    }e[200001];
    int n,m,tot,size;
    int v[100001];
    int b[100001];
    int head[100001];
    int root[100001];
    int st[100001][17];
    int dep[100001];
    struct node{
        int size,ls,rs;
        node(){
            size=ls=rs=0;
        }
    }t[100001*30];
    void build(int l,int r,int &now){
        now=++tot;
        if(l==r)return;
        int mid=l+r>>1;
        build(l,mid,t[now].ls);
        build(mid+1,r,t[now].rs);
    }
    void addtree(int l,int r,int &now,int pre,int x){
        now=++tot;
        t[now]=t[pre];
        t[now].size++;
        if(l==r)return;
        int mid=l+r>>1;
        if(x<=mid)addtree(l,mid,t[now].ls,t[pre].ls,x);
        else addtree(mid+1,r,t[now].rs,t[pre].rs,x);
    }
    int query(int l,int r,int now1,int now2,int pre1,int pre2,int k){
        if(l==r)return b[l];
        int lsize=t[t[now1].ls].size+t[t[now2].ls].size-t[t[pre1].ls].size-t[t[pre2].ls].size;
        int mid=l+r>>1;
        if(k<=lsize)return query(l,mid,t[now1].ls,t[now2].ls,t[pre1].ls,t[pre2].ls,k);
        else return query(mid+1,r,t[now1].rs,t[now2].rs,t[pre1].rs,t[pre2].rs,k-lsize);
    }
    void dfs(int x,int fa){
        st[x][0]=fa;
        dep[x]=dep[fa]+1;
        int num=lower_bound(b+1,b+size+1,v[x])-b;
        addtree(1,size,root[x],root[fa],num);
        for(int i=head[x];i;i=e[i].next){
            int u=e[i].to;
            if(u==fa)continue;
            dfs(u,x);
        }
    }
    void getst(){
        for(int k=1;k<=16;k++){
            for(int i=1;i<=n;i++){
                st[i][k]=st[st[i][k-1]][k-1];
            }
        }
    }
    int getlca(int x,int y){
        if(x==y)return x;
        if(dep[x]>dep[y])swap(x,y);
        for(int k=16;k>=0;k--){
            if(dep[st[y][k]]>=dep[x]){
                y=st[y][k];
            }
        }
        if(x==y)return x;
        for(int k=16;k>=0;k--){
            if(st[x][k]!=st[y][k]){
                x=st[x][k];
                y=st[y][k];
            }
        }
        return st[x][0];
    }
    int ask(int x,int y,int k){
        int lca=getlca(x,y);
        int ans=query(1,size,root[x],root[y],root[lca],root[st[lca][0]],k);
        return ans;
    }
    inline void addedge(int x,int y){
        e[++tot].to=y;e[tot].next=head[x];head[x]=tot;
    }
    int main(){
        n=read();m=read();
        for(int i=1;i<=n;i++){
            v[i]=read();b[i]=v[i];
        }
        sort(b+1,b+n+1);
        size=unique(b+1,b+n+1)-b-1;
        for(int i=1;i<=n-1;i++){
            int x=read(),y=read();
            addedge(x,y);addedge(y,x);
        }
        tot=0;
        build(1,size,root[0]);
        dfs(1,0);
        getst();
        int ans=0;
        for(int i=1;i<=m;i++){
            int x=read(),y=read(),k=read();
            ans=ask(x^ans,y,k);
            printf("%d
    ",ans);
        }
        return 0;
    }
  • 相关阅读:
    pytorch常用函数
    检测(2):开始训练
    gcc的替换,很有用
    detection-pytorch环境配置的遇到的坑
    一些有用的ubuntu命令总结---长期更新
    如何用gdb工具进行调试
    检测(1)从0到1
    检测
    pytorch遇到的问题---不定期更新
    假名快速记忆
  • 原文地址:https://www.cnblogs.com/stone41123/p/7581242.html
Copyright © 2011-2022 走看看