BZOJ1036 [ZJOI2008]树的统计Count 树链剖分

来源：互联网发布：js控制元素显示隐藏编辑：程序博客网时间：2024/04/20 09:04

一道很裸的树链剖分题目，将树轻重链剖分后用线段树维护重链，记录和与最大值即可。

对着标程敲……顺便精简了一下。感觉这样就已经很短了。

1A！好久没1A了……虽然是照标程改的不过还是很有成就感的。

//BZOJ1036#include<cstdio>#include<algorithm>#include<cstring>#include<vector>#include<utility>using namespace std;const int MAXN=30010;const int INF=1e9;vector<int> graph[MAXN];int n,in1,in2,w[MAXN],q;int p[MAXN],size[MAXN],son[MAXN],dep[MAXN],top[MAXN],tid[MAXN],tot;int l[MAXN<<2],r[MAXN<<2];pair<int,int> tree[MAXN<<2];char a[20];pair<int,int> operator+(const pair<int,int> &s1,const pair<int,int> &s2){return make_pair(s1.first+s2.first,max(s1.second,s2.second));}void DFS(int x){size[x]=1,son[x]=0;int len=graph[x].size();for(int i=0;i<len;i++){int t=graph[x][i];if(t!=p[x]){p[t]=x,dep[t]=dep[x]+1;DFS(t);size[x]+=size[t];if(size[t]>size[son[x]]) son[x]=t;}}}void build_tree(int x,int anc){tid[x]=++tot,top[x]=anc,tree[x].second=-INF;int len=graph[x].size();if(son[x]!=0) build_tree(son[x],anc);for(int i=0;i<len;i++){int t=graph[x][i];if(t!=p[x]&&t!=son[x]) build_tree(t,t);}}void MakeSegTree(int x,int ll,int rr){l[x]=ll,r[x]=rr;if(ll==rr) return;int m=(l[x]+r[x])>>1;MakeSegTree(x<<1,ll,m);MakeSegTree(x<<1|1,m+1,rr);}void SegChange(int x,int pos,int tar){if(l[x]==r[x]) {tree[x]=make_pair(tar,tar);return;}int m=(l[x]+r[x])>>1;if(pos<=m) SegChange(x<<1,pos,tar);else SegChange(x<<1|1,pos,tar);tree[x]=tree[x<<1]+tree[x<<1|1];}pair<int,int> SegQuery(int x,int ll,int rr){if(l[x]==ll&&r[x]==rr) return tree[x];int m=(l[x]+r[x])>>1;if(rr<=m) return SegQuery(x<<1,ll,rr);else if(ll>m) return SegQuery(x<<1|1,ll,rr);else return SegQuery(x<<1,ll,m)+SegQuery(x<<1|1,m+1,rr);}pair<int,int> TreeQuery(int p1,int p2){int t1=top[p1],t2=top[p2];pair<int,int> con=make_pair(0,-INF);while(t1!=t2){if(dep[top[p1]]<dep[top[p2]]) {swap(p1,p2);swap(t1,t2);}con=con+SegQuery(1,tid[t1],tid[p1]);p1=p[t1],t1=top[p1];}if(dep[p1]<dep[p2]) swap(p1,p2);con=con+SegQuery(1,tid[p2],tid[p1]);return con;}int main(){scanf("%d",&n);for(int i=1;i<n;i++){scanf("%d%d",&in1,&in2);graph[in1].push_back(in2);graph[in2].push_back(in1);}for(int i=1;i<=n;i++) scanf("%d",&w[i]);DFS(1);build_tree(1,1);MakeSegTree(1,1,n);for(int i=1;i<=n;i++) SegChange(1,tid[i],w[i]);scanf("%d",&q);for(int cas=1;cas<=q;cas++){scanf("%s%d%d",a,&in1,&in2);if(a[0]=='C') SegChange(1,tid[in1],in2);else if(a[1]=='M') printf("%d\n",TreeQuery(in1,in2).second);else printf("%d\n",TreeQuery(in1,in2).first);}return 0;}

0 0