标签：cnt HYSBZ1036 剖分 int tree son dfs2 ZJOI2008 include

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

  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 }

标签：cnt,HYSBZ1036,剖分,int,tree,son,dfs2,ZJOI2008,include
来源： https://www.cnblogs.com/yhxnoerror/p/16435754.html

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