BZOJ 1036([ZJOI2008]树的统计Count-树链剖分[成熟版])

1036: [ZJOI2008]树的统计Count

Time Limit: 10 Sec Memory Limit: 162 MB

Submit: 3205 Solved: 1335

[Submit][Status]

Description

一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。 我们将以下面的形式来要求你对这棵树完成一些操作： I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 III. QSUM u v: 询问从点u到点v的路径上的节点的权值和 注意：从点u到点v的路径上的节点包括u和v本身

Input

输入的第一行为一个整数n，表示节点的个数。 接下来n – 1行，每行2个整数a和b，表示节点a和节点b之间有一条边相连。 接下来n行，每行一个整数，第i行的整数wi表示节点i的权值。 接下来1行，为一个整数q，表示操作的总数。 接下来q行，每行一个操作，以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。 对于100％的数据，保证1<=n<=30000，0<=q<=200000；中途操作中保证每个节点的权值w在-30000到30000之间。

Output

对于每个“QMAX”或者“QSUM”的操作，每行输出一个整数表示要求输出的结果。

Sample Input

4

1 2

2 3

4 1

4 2 1 3

12

QMAX 3 4

QMAX 3 3

QMAX 3 2

QMAX 2 3

QSUM 3 4

QSUM 2 1

CHANGE 1 5

QMAX 3 4

CHANGE 3 6

QMAX 3 4

QMAX 2 4

QSUM 3 4

Sample Output

4

1

2

2

10

6

5

6

5

16

在yzc的RP++模板下：

我完成了成熟版的树链剖分

本次更新的主要内容：

1.趋向于把值存在点上

2.引入了dfs序的概念

3.成功的只建一棵线段树，从此和可持久化说再见

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<functional>
#include<iostream>
#include<cmath>
#include<cctype>
#include<ctime>
using namespace std;
#define For(i,n) for(int i=1;i<=n;i++)
#define Fork(i,k,n) for(int i=k;i<=n;i++)
#define Rep(i,n) for(int i=0;i<n;i++)
#define ForD(i,n) for(int i=n;i;i--)
#define RepD(i,n) for(int i=n;i>=0;i--)
#define Forp(x) for(int p=pre[x];p;p=next[p])
#define Lson (x<<1)
#define Rson ((x<<1)+1)
#define MEM(a) memset(a,0,sizeof(a));
#define MEMI(a) memset(a,127,sizeof(a));
#define MEMi(a) memset(a,128,sizeof(a));
#define INF (2139062143)
#define F (100000007)
#define MAXN (30000+10)
#define MAXM (200000+10)
long long mul(long long a,long long b){return (a*b)%F;}
long long add(long long a,long long b){return (a+b)%F;}
long long sub(long long a,long long b){return (a-b+(a-b)/F*F+F)%F;}
typedef long long ll;
int n,m;
ll sum[MAXN<<3]={0},maxv[MAXN<<3]={0};
void update(int x){sum[x]=sum[Lson]+sum[Rson],maxv[x]=max(maxv[Lson],maxv[Rson]);}
void change(int x,int l,int r,int L,int c)
{
if (l==r) {sum[x]=maxv[x]=c;return;}
int m=l+r>>1;
if (L<=m) change(Lson,l,m,L,c);
else change(Rson,m+1,r,L,c);
update(x);
}
ll qurrysum(int x,int l,int r,int L,int R)
{
if (L<=l&&r<=R) return sum[x];
int m=l+r>>1;
ll ans=0;
if (L<=m) ans+=qurrysum(Lson,l,m,L,R);
if (m<R) ans+=qurrysum(Rson,m+1,r,L,R);
return ans;
}
ll qurrymax(int x,int l,int r,int L,int R)
{
if (L<=l&&r<=R) return maxv[x];
int m=l+r>>1;
ll ans=-INF;
if (L<=m) ans=max(ans,qurrymax(Lson,l,m,L,R));
if (m<R) ans=max(ans,qurrymax(Rson,m+1,r,L,R));
return ans;
}
int edge[MAXN*2],pre[MAXN]={0},next[MAXN*2]={0},size=0;
void addedge(int u,int v)
{
edge[++size]=v;
next[size]=pre[u];
pre[u]=size;
}
void addedge2(int u,int v){addedge(u,v),addedge(v,u);}
int dep[MAXN]={0},siz[MAXN]={0},opt[MAXN]={0},father[MAXN]={0};
void dfs(int x)
{
siz[x]=1;
dep[x]=dep[father[x]]+1;
Forp(x)
{
int &v=edge[p];
if (v^father[x])
{
father[v]=x;
dfs(v);
siz[x]+=siz[v];
if (siz[v]>siz[opt[x]]) opt[x]=v;
}
}
}
int top[MAXN],w[MAXN],place[MAXN],tim=0;
void dfs2(int x,int chain)
{
top[x]=chain;
place[x]=++tim;
change(1,1,n,place[x],w[x]);
if (opt[x]) dfs2(opt[x],chain);else return;
Forp(x)
{
int &v=edge[p];
if (v^father[x]&&v^opt[x])
{
dfs2(v,v);
}
}
}
ll lca_sum(int u,int v)
{
ll ans=0;
while (top[u]^top[v])
{
if (dep[top[u]]<dep[top[v]]) swap(u,v);
ans+=qurrysum(1,1,n,place[top[u]],place[u]);
u=father[top[u]];
}
if (dep[u]<dep[v]) swap(u,v);
ans+=qurrysum(1,1,n,place[v],place[u]);
return ans;
}
ll lca_max(int u,int v)
{
ll ans=-INF;
while (top[u]^top[v])
{
if (dep[top[u]]<dep[top[v]]) swap(u,v);
ans=max(ans,qurrymax(1,1,n,place[top[u]],place[u]));
u=father[top[u]];
}
if (dep[u]<dep[v]) swap(u,v);
ans=max(ans,qurrymax(1,1,n,place[v],place[u]));
return ans;
}
char c[10];
int main()
{
//	freopen("bzoj1036.in","r",stdin);
scanf("%d",&n);
/*
For(i,n)
{
int p,l,r,c;
cin>>p>>l>>r;
if (p==1) change(1,1,n,l,r);
if (p==2) cout<<qurrysum(1,1,n,l,r)<<endl;
if (p==3) cout<<qurrymax(1,1,n,l,r)<<endl;

}*/
For(i,n-1)
{
int u,v;
scanf("%d%d",&u,&v);
addedge2(u,v);
}
For(i,n) scanf("%d",&w[i]);
dfs(1);
dfs2(1,1);
scanf("%d",&m);
For(i,m)
{
int a,b;
scanf("%s%d%d",c,&a,&b);
switch(c[1])
{
case'H':change(1,1,n,place[a],b);break;
case'S':printf("%lld\n",lca_sum(a,b));break;
case'M':printf("%lld\n",lca_max(a,b));
}
}
return 0;
}