[BZOJ1036][ZJOI2008]树的统计Count

最新推荐文章于 2019-08-03 10:19:00 发布

Zvezda_

最新推荐文章于 2019-08-03 10:19:00 发布

阅读量288

点赞数

CC 4.0 BY-SA版权

分类专栏： BZOJ 图论_树链剖分数据结构_线段树图论

本文链接：https://blog.youkuaiyun.com/Zvezda_/article/details/46948001

BZOJ 同时被 3 个专栏收录

159 篇文章

订阅专栏

图论

29 篇文章

订阅专栏

数据结构_线段树

13 篇文章

订阅专栏

本文深入探讨了树链剖分和线段树的结合使用，通过一个具体的编程实例展示了如何解决复杂的查询更新问题。文章详细介绍了算法实现步骤，并提供了AC代码供读者参考。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

原题地址

树链剖分+线段树

AC code:

#include <cstdio>
#include <vector>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=30010;
const int INF=1<<28;
int n,q,tot,c1,c2;
int size[N],num[N],bl[N],top[N],last[N];
vector<int> G[N];

struct nod{
    int l,r,s,m;
    nod *lc,*rc;
}pool[N<<2];

struct Segtree{
    nod *root;

    void build(nod **p,int L,int R){
        *p=&pool[tot++];(*p)->l=L;(*p)->r=R;
        if(L==R){
            (*p)->lc=(*p)->rc=NULL;
            return ;
        }
        int M=(L+R)>>1;
        build(&(*p)->lc,L,M);
        build(&(*p)->rc,M+1,R);
    }
    void modify(nod *p,int pos,int k){
        if(p->l==p->r){
            p->s=p->m=k;
            return ;
        }
        int M=(p->l+p->r)>>1;
        if(pos<=M) modify(p->lc,pos,k);
        else modify(p->rc,pos,k);
        p->s=p->lc->s+p->rc->s;
        p->m=max(p->lc->m,p->rc->m);
    }
    int getmax(nod *p,int L,int R){
        if(p->l==L&&p->r==R) return p->m;
        int M=(p->l+p->r)>>1;
        if(R<=M) return getmax(p->lc,L,R);
        else if(L>M) return getmax(p->rc,L,R);
        else return max(getmax(p->lc,L,M),getmax(p->rc,M+1,R));
    }
    int getsum(nod *p,int L,int R){
        if(p->l==L&&p->r==R) return p->s;
        int M=(p->l+p->r)>>1;
        if(R<=M) return getsum(p->lc,L,R);
        else if(L>M) return getsum(p->rc,L,R);
        else return getsum(p->lc,L,M)+getsum(p->rc,M+1,R);
    }
}T;

void DFS(int pre,int p){
    size[p]=1;
    for(int i=0;i<G[p].size();i++){
        int q=G[p][i];
        if(q==pre) continue;
        DFS(p,q);
        size[p]+=size[q];
    }
}

void cut(int pre,int p,int number,int belong){
    num[p]=number;bl[p]=belong;
    int q=0,maxs=0;
    for(int i=0;i<G[p].size();i++){
        int next=G[p][i];
        if(next==pre) continue;
        if(size[next]>maxs){
            q=G[p][i];
            maxs=size[G[p][i]];
        }
    }
    if(q) cut(p,q,++c1,c2);
    for(int i=0;i<G[p].size();i++){
        int next=G[p][i];
        if(next==pre||next==q) continue;
        else{
            top[++c2]=next;
            last[c2]=p;
            cut(p,next,++c1,c2);
        }
    }
}

void swap(int *x,int *y){
    int t=*x;*x=*y;*y=t;
}

void qmax(int u,int v){
    int ans=-INF;
    while(bl[u]!=bl[v]){
        if(bl[u]>bl[v]) swap(&u,&v);
        ans=max(ans,T.getmax(T.root,num[top[bl[v]]],num[v]));
        v=last[bl[v]];
    }
    if(num[u]>num[v]) swap(&u,&v);
    ans=max(ans,T.getmax(T.root,num[u],num[v]));
    printf("%d\n",ans);
}

void qsum(int u,int v){
    int ans=0;
    while(bl[u]!=bl[v]){
        if(bl[u]>bl[v]) swap(&u,&v);
        ans+=T.getsum(T.root,num[top[bl[v]]],num[v]);
        v=last[bl[v]];
    }
    if(num[u]>num[v]) swap(&u,&v);
    ans+=T.getsum(T.root,num[u],num[v]);
    printf("%d\n",ans);
}

int main(){
    scanf("%d",&n);
    T.build(&T.root,1,n);
    for(int i=1;i<n;i++){
        int a,b;
        scanf("%d%d",&a,&b);
        G[a].push_back(b);
        G[b].push_back(a);
    }
    DFS(0,1);
    top[1]=1;
    cut(0,1,++c1,++c2);
    for(int i=1;i<=n;i++){
        int t;
        scanf("%d",&t);
        T.modify(T.root,num[i],t);
    }
    scanf("%d",&q);
    for(int i=1;i<=q;i++){
        int  a,b;
        char ord[11];
        scanf("%s%d%d",ord,&a,&b);
        if(!strcmp(ord,"CHANGE")) T.modify(T.root,num[a],b);
        else if(!strcmp(ord,"QMAX")) qmax(a,b);
        else qsum(a,b);
    }

    return 0;
}