bzoj 1036: [ZJOI2008]树的统计Count

1036: [ZJOI2008]树的统计Count

Time Limit: 10 Sec Memory Limit: 162 MB

Submit: 16200 Solved: 6612

[Submit][Status][Discuss]

Description

　　一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成

一些操作： I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I

II. 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

【分析】

忘记设置初始值WA了好几次2333….

【代码】

//bzoj
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define inf 1e9+7
#define M(a) memset(a,0,sizeof a)
#define fo(i,j,k) for(i=j;i<=k;++i)
using namespace std;
const int mxn=300005;
int n,m,tot,cnt;
int head[mxn],w[mxn],pos[mxn];
struct edge {int to,next;} f[mxn<<1];
struct tree
{
int fa,son,s,e,top,sz,dep;
}e[mxn];
struct line
{
int l,r,mx,sum;
}t[mxn<<2];
inline void add(int u,int v)
{
f[++cnt].to=v,f[cnt].next=head[u],head[u]=cnt;
}
inline void dfs1(int u)
{
e[u].sz=1;
for(int i=head[u];i;i=f[i].next)
{
int v=f[i].to;
if(v==e[u].fa) continue;
e[v].fa=u;
e[v].dep=e[u].dep+1;
dfs1(v);
e[u].sz+=e[v].sz;
if(e[v].sz>e[e[u].son].sz)
e[u].son=v;
}
}
inline void dfs2(int u,int top)
{
e[u].top=top;
e[u].s=++tot;
pos[tot]=u;
if(e[u].son)
{
dfs2(e[u].son,top);
for(int i=head[u];i;i=f[i].next)
{
int v=f[i].to;
if(v!=e[u].fa && v!=e[u].son)
dfs2(v,v);
}
}
e[u].e=tot;
}
inline void update(int num)
{
t[num].mx=max(t[num<<1].mx,t[num<<1|1].mx);
t[num].sum=t[num<<1].sum+t[num<<1|1].sum;
}
inline void build(int num,int l,int r)
{
t[num].l=l,t[num].r=r;
if(l==r)
{
t[num].mx=t[num].sum=w[pos[l]];
return;
}
int mid=l+r>>1;
build(num<<1,l,mid);
build(num<<1|1,mid+1,r);
update(num);
}
inline void change(int num,int L,int R,int c)
{
if(L<=t[num].l && t[num].r<=R)
{
t[num].mx=t[num].sum=c;
return;
}
if(L<=t[num<<1].r) change(num<<1,L,R,c);
if(R>=t[num<<1|1].l) change(num<<1|1,L,R,c);
update(num);
}
inline int Qsum(int num,int L,int R)
{
if(L<=t[num].l && t[num].r<=R)
return t[num].sum;
int ans=0;
if(L<=t[num<<1].r) ans+=Qsum(num<<1,L,R);
if(R>=t[num<<1|1].l) ans+=Qsum(num<<1|1,L,R);
return ans;
}
inline int Qmax(int num,int L,int R)
{
if(L<=t[num].l && t[num].r<=R)
return t[num].mx;
int mx=-inf;
if(L<=t[num<<1].r) mx=max(mx,Qmax(num<<1,L,R));
if(R>=t[num<<1|1].l) mx=max(mx,Qmax(num<<1|1,L,R));
return mx;
}
inline int Fsum(int x,int y)
{
int ans=0;
int f1=e[x].top,f2=e[y].top;
while(f1!=f2)
{
if(e[f1].dep<e[f2].dep)
swap(x,y),swap(f1,f2);
ans+=Qsum(1,e[f1].s,e[x].s);
x=e[f1].fa;
f1=e[x].top;
}
if(e[x].dep<e[y].dep)
ans+=Qsum(1,e[x].s,e[y].s);
else
ans+=Qsum(1,e[y].s,e[x].s);
return ans;
}
inline int Fmax(int x,int y)
{
int ans=-inf;
int f1=e[x].top,f2=e[y].top;
while(f1!=f2)
{
if(e[f1].dep<e[f2].dep)
swap(x,y),swap(f1,f2);
ans=max(ans,Qmax(1,e[f1].s,e[x].s));
x=e[f1].fa;
f1=e[x].top;
}
if(e[x].dep<e[y].dep)
ans=max(ans,Qmax(1,e[x].s,e[y].s));
else
ans=max(ans,Qmax(1,e[y].s,e[x].s));
return ans;
}
int main()
{
//  freopen("bzoj_1036.in","r",stdin);
//  freopen("bzoj_1036.out","w",stdout);
int i,j,u,v,x,y;
scanf("%d",&n);
fo(i,2,n)
{
scanf("%d%d",&u,&v);
add(u,v),add(v,u);
}
fo(i,1,n) scanf("%d",&w[i]);
dfs1(1);
dfs2(1,1);
build(1,1,n);
scanf("%d",&m);
while(m--)
{
char opt[10];
scanf("%s",opt);
scanf("%d%d",&u,&v);
if(opt[0]=='C')
change(1,e[u].s,e[u].s,v);
else if(opt[1]=='S')
printf("%d\n",Fsum(u,v));
else
printf("%d\n",Fmax(u,v));
}
return 0;
}