bzoj1036: [ZJOI2008]树的统计Count
第一道树链剖分。。。
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
#define rep(i,n) for(int i=1;i<=n;i++)
#define clr(x,c) memset(x,c,sizeof(x))
#define adde(u,v) add(u,v) ,add(v,u)
#define qwq(x) for(edge *o=head[x];o;o=o->next)
#define lson l,m,x<<1
#define rson m+1,r,x<<1|1
int read(){
int x=0;char c=getchar();bool f=true;
while(!isdigit(c)) {
if(c=='-') f=false;c=getchar();
}
while(isdigit(c)) x=x*10+c-'0',c=getchar();
return f?x:-x;
}
const int nmax=30005;
const int inf=0x7f7f7f7f;
struct edge{
int to;
edge *next;
};
edge edges[nmax<<1],*pt=edges,*head[nmax];
bool vis[nmax];
int id[nmax],idx[nmax<<2],fa[nmax],size[nmax],son[nmax],dep[nmax],tp[nmax],w[nmax],n;
char ch[100];
struct node{
int maxn,sum;
};
node tree[nmax<<2];
void add(int s,int t){
pt->to=t;pt->next=head[s];head[s]=pt++;
}
void dfs(int x){
vis[x]=1;size[x]=1;
qwq(x){
int to=o->to;
if(!vis[to]){
dep[to]=dep[x]+1;fa[to]=x;dfs(to);
size[x]+=size[to];
if(!son[x]||size[to]>size[son[x]]) son[x]=to;
}
}
}
void Dfs(int x,int top){
id[x]=++id[0];idx[id[0]]=x;tp[x]=top;
if(son[x]) Dfs(son[x],top);
qwq(x) if(!id[o->to]) Dfs(o->to,o->to);
}
void pushup(int x){
node &o=tree[x];node &tl=tree[x<<1];node &tr=tree[x<<1|1];
o.maxn=max(tl.maxn,tr.maxn);o.sum=tl.sum+tr.sum;
}
void build(int l,int r,int x){
node &o=tree[x];
if(l==r) {
o.maxn=o.sum=w[idx[l]];return ;
}
int m=(l+r)>>1;build(lson);build(rson);pushup(x);
}
void update(int p,int add,int l,int r,int x){
if(l==r) {
tree[x].maxn=tree[x].sum=add;return ;
}
int m=(l+r)>>1;
p<=m?update(p,add,lson):update(p,add,rson);pushup(x);
}
int querysum(int tl,int tr,int l,int r,int x){
if(tl<=l&&tr>=r) return tree[x].sum;
int m=(l+r)>>1;int ans=0;
if(tl<=m) ans+=querysum(tl,tr,lson);
if(tr>m) ans+=querysum(tl,tr,rson);
return ans;
}
int querymax(int tl,int tr,int l,int r,int x){
if(tl<=l&&tr>=r) return tree[x].maxn;
int m=(l+r)>>1;int ans=-inf;
if(tl<=m) ans=max(ans,querymax(tl,tr,lson));
if(tr>m) ans=max(ans,querymax(tl,tr,rson));
return ans;
}
int qsum(int a,int b){
int ans=0;
while(tp[a]!=tp[b]){
if(dep[tp[a]]>dep[tp[b]]) swap(a,b);
ans+=querysum(id[tp[b]],id[b],1,n,1);
b=fa[tp[b]];
}
if(dep[a]>dep[b]) swap(a,b);
ans+=querysum(id[a],id[b],1,n,1);return ans;
}
int qmax(int a,int b){
int ans=-inf;
while(tp[a]!=tp[b]){
if(dep[tp[a]]>dep[tp[b]]) swap(a,b);
ans=max(ans,querymax(id[tp[b]],id[b],1,n,1));
b=fa[tp[b]];
}
if(dep[a]>dep[b]) swap(a,b);
ans=max(ans,querymax(id[a],id[b],1,n,1));return ans;
}
void test(int l,int r,int x){
if(l==r) printf("%d ",tree[x].maxn);
else{
int m=(l+r)>>1;test(lson);test(rson);
}
}
int main(){
n=read();
rep(i,n-1) {
int s=read(),t=read();adde(s,t);
}
/*rep(i,n){
for(edge *o=head[i];o;o=o->next) printf("%d ",o->to);
printf("\n");
}*/
clr(vis,0);clr(son,0);dep[1]=0;
clr(id,0);clr(idx,0);id[0]=0;
dfs(1);Dfs(1,1);
/*rep(i,n){
printf("%d:%d %d %d %d\n",i,fa[i],son[i],size[i],dep[i]);
}*/
rep(i,n) w[i]=read();
// rep(i,n) printf("%d ",id[i]);
build(1,n,1);//test(1,n,1);
int m=read();
rep(i,m){
scanf("%s",ch);int a=read(),b=read();
if(ch[0]=='C') update(id[a],b,1,n,1);
else if(ch[1]=='M') printf("%d\n",qmax(a,b));
else printf("%d\n",qsum(a,b));
}
return 0;
}
1036: [ZJOI2008]树的统计Count
Time Limit: 10 Sec Memory Limit: 162 MBSubmit: 12514 Solved: 5045
[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
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
1
2
2
10
6
5
6
5
16
