HYSBZ1036 [ZJOI2008]树的统计(树链剖分)

2022/7/1 23:23:44

本文主要是介绍HYSBZ1036 [ZJOI2008]树的统计(树链剖分),对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

将树通过树链剖分转化成线性序列,用线段树维护最值,和值即可。

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<algorithm>
  5 using namespace std;
  6 const int maxn=30005;
  7 int n,m;
  8 int head[maxn],to[maxn<<1],nxt[maxn<<1],tot;
  9 int fa[maxn],dep[maxn],size[maxn],son[maxn];
 10 int top[maxn],id[maxn],rev[maxn];
 11 int w[maxn],cnt;
 12 int Max,Sum;
 13 
 14 struct node{
 15     int l,r,sum,mx;
 16 }tree[maxn*4];
 17 
 18 int readin(){//注意:本题c++读入会超时 
 19     char c;
 20     int sign=1;
 21     while((c=getchar())<'0'||c>'9')
 22         if(c=='-')
 23             sign=-1;
 24     int res=c-'0';
 25     while((c=getchar())>='0'&&c<='9')
 26         res=res*10+c-'0';
 27     return res*sign;
 28 }
 29 
 30 void add(int u,int v){
 31     nxt[++tot]=head[u];
 32     head[u]=tot;
 33     to[tot]=v;
 34 }
 35 
 36 void dfs1(int u,int f){
 37     fa[u]=f,dep[u]=dep[f]+1,size[u]=1;
 38     for(int i=head[u];i;i=nxt[i]){
 39         int v=to[i];
 40         if(v==f) continue;
 41         dfs1(v,u);
 42         size[u]+=size[v];
 43         if(size[v]>size[son[u]]) son[u]=v;
 44     }
 45 }
 46 
 47 void dfs2(int u,int t){
 48     top[u]=t;
 49     id[u]=++cnt;
 50     rev[cnt]=u;
 51     if(!son[u]) return ;
 52     dfs2(son[u],t);//优先沿着重儿子dfs
 53     for(int i=head[u];i;i=nxt[i]){
 54         int v=to[i];
 55         if(v!=fa[u] && v!=son[u]) dfs2(v,v);
 56     } 
 57 }
 58 
 59 void pushup(int k){
 60     tree[k].mx=max(tree[k<<1].mx,tree[k<<1|1].mx);
 61     tree[k].sum=tree[k<<1].sum+tree[k<<1|1].sum;
 62 }
 63 
 64 void build(int k,int l,int r){
 65     tree[k].l=l,tree[k].r=r;
 66     int mid=(l+r)>>1;
 67     if(l==r){tree[k].sum=tree[k].mx=w[rev[l]]; return ;}
 68     build(k<<1,l,mid);
 69     build(k<<1|1,mid+1,r);
 70     pushup(k);
 71 }
 72 
 73 void change(int k,int x,int y){
 74     if(tree[k].l==tree[k].r&&tree[k].l==x){
 75         tree[k].mx=tree[k].sum=y;
 76         return ;
 77     }
 78     int mid=(tree[k].l+tree[k].r)>>1;
 79     if(x<=mid) change(k<<1,x,y);
 80     else change(k<<1|1,x,y);
 81     pushup(k);
 82 }
 83 
 84 void query(int k,int l,int r){
 85     if(tree[k].l>=l&&tree[k].r<=r){//找到该区间
 86         Max=max(Max,tree[k].mx);
 87         Sum+=tree[k].sum;
 88         return;
 89     } 
 90     int mid=(tree[k].l+tree[k].r)>>1;
 91     if(l<=mid) query(k<<1,l,r);
 92     if(r>mid) query(k<<1|1,l,r);
 93 }
 94 
 95 void ask(int u,int v){//求u,v之间的最值或和值 
 96     while(top[u]!=top[v]){
 97         if(dep[top[u]]<dep[top[v]]) swap(u,v);
 98         query(1,id[top[u]],id[u]);
 99         u=fa[top[u]];
100     }
101     if(dep[u]>dep[v]) swap(u,v);
102     query(1,id[u],id[v]);
103 }
104 
105 int main(){
106     int x,y; char str[10];
107     n=readin();
108     for(int i=1;i<n;i++){
109         x=readin();y=readin();
110         add(x,y);add(y,x);
111     }
112     for(int i=1;i<=n;i++) cin>>w[i];
113     dfs1(1,0);
114     dfs2(1,1);
115     build(1,1,cnt);//创建线段树
116     m=readin();
117     for(int i=1;i<=m;i++){
118         scanf("%s",str);
119         x=readin();y=readin();
120         if(str[0]=='C')
121             change(1,id[x],y);
122         else{
123             Sum=0; Max=-0x3f3f3f3f;
124             ask(x,y);
125             if(str[1]=='M') printf("%d\n",Max);
126             else printf("%d\n",Sum);    
127         }    
128     }
129     return 0;
130 }

 



这篇关于HYSBZ1036 [ZJOI2008]树的统计(树链剖分)的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程