bzoj 1036 洛谷 p2590 [ZJOI2008]树的统计

题目描述

一棵树上有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本身

输入输出格式

输入格式：
输入文件的第一行为一个整数n，表示节点的个数。

接下来n – 1行，每行2个整数a和b，表示节点a和节点b之间有一条边相连。

接下来一行n个整数，第i个整数wi表示节点i的权值。

接下来1行，为一个整数q，表示操作的总数。

接下来q行，每行一个操作，以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。

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

输入输出样例

输入样例#1：
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：
4
1
2
2
10
6
5
6
5
16
说明

对于100％的数据，保证1<=n<=30000，0<=q<=200000；中途操作中保证每个节点的权值w在-30000到30000之间。

#include <cstdio>#include <iostream>#include <cstring>#include <algorithm>#define int long long#define ls id<<1#define rs id<<1|1using namespace std;const int N=100005;int n,m,r,p,tot,dcnt,q;int w[N],head[N*2],mp[N];struct NODE{    int to,nex;}ed[N*2];struct node{    int fa,son,sz,dep,top,s,e;}tr[N];//树链剖分struct Segtree{   int l,r;   int sum,lazy,maxx;}tree[4*N];inline void addedge(int x,int y){    ++tot;    ed[tot].nex=head[x];    head[x]=tot;    ed[tot].to=y;}inline int read(){    int x=0,f=1;    char ch=getchar();    while(ch<'0'||ch>'9') {if(ch=='-')f=-1;ch=getchar();}    while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}    return f*x;}void dfs1(int u){    tr[u].sz=1;    for(int i=head[u];i;i=ed[i].nex)       if(ed[i].to!=tr[u].fa){             tr[ed[i].to].fa=u;             tr[ed[i].to].dep=tr[u].dep+1;             dfs1(ed[i].to);             tr[u].sz+=tr[ed[i].to].sz;             if(tr[ed[i].to].sz>tr[tr[u].son].sz){                  tr[u].son=ed[i].to;             }    }}void dfs2(int u,int top){    tr[u].top=top;    tr[u].s=++dcnt;    mp[dcnt]=u;    if(tr[u].son){        dfs2(tr[u].son,top);        for(int i=head[u];i;i=ed[i].nex)         if(ed[i].to!=tr[u].fa&&ed[i].to!=tr[u].son)         dfs2(ed[i].to,ed[i].to);            }    tr[u].e=dcnt;}inline void update(int b){    tree[b].sum=tree[b*2].sum+tree[b*2+1].sum;    tree[b].maxx=max(tree[b*2].maxx,tree[b*2+1].maxx);}void build(int b,int l,int r){    tree[b].l=l;    tree[b].r=r;    if(l==r){        tree[b].sum=tree[b].maxx=w[mp[l]];        return ;    }    int mid=l+r>>1;    build(b*2,l,mid);    build(b*2+1,mid+1,r);    update(b);}int qsum(int id,int l,int r){    if(tree[id].l>r || tree[id].r<l)return 0;    if(tree[id].l>=l && tree[id].r<=r)return tree[id].sum;    return (qsum(id*2,l,r)+qsum(id*2+1,l,r));}int qmax(int id,int l,int r){    int ans=-1e9;    if(tree[id].l>r || tree[id].r<l)return -1e9;    if(tree[id].l>=l && tree[id].r<=r)return max(ans,tree[id].maxx);    return max(qmax(id*2,l,r),qmax(id*2+1,l,r));}int query(int x,int y){    int f1=tr[x].top;    int f2=tr[y].top;    int ans=0;    while(f1!=f2){        if(tr[f1].dep<tr[f2].dep){            swap(x,y);            swap(f1,f2);        }        ans+=qsum(1,tr[f1].s,tr[x].s);        x=tr[f1].fa;        f1=tr[x].top;    }    if(tr[x].dep>tr[y].dep) swap(x,y);        return ans=ans+qsum(1,tr[x].s,tr[y].s);}int find(int x,int y){    int f1=tr[x].top;    int f2=tr[y].top;    int ans=-10000000;    while(f1!=f2){        if(tr[f1].dep<tr[f2].dep){            swap(x,y);            swap(f1,f2);        }        ans=max(ans,qmax(1,tr[f1].s,tr[x].s));        x=tr[f1].fa;        f1=tr[x].top;    }    if(tr[x].dep<tr[y].dep)     ans=max(ans,qmax(1,tr[x].s,tr[y].s));    else       ans=max(ans,qmax(1,tr[y].s,tr[x].s));    return ans;}void change(int b,int x,int y,int z){    if(y<tree[b].l||x>tree[b].r) return ;    if(tree[b].l==tree[b].r){        tree[b].maxx=z;        tree[b].sum=z;        return ;    }    change(b*2,x,y,z);    change(b*2+1,x,y,z);    update(b);}main(){    n=read();    for(int i=1;i<n;i++){      int x,y;      x=read();      y=read();      addedge(x,y);      addedge(y,x);    }    for(int i=1;i<=n;i++)  w[i]=read();    dfs1(1);    dfs2(1,1);    build(1,1,n);    q=read();    for(int i=1;i<=q;i++){        string mode;        int x,y,z;        cin>>mode;        if(mode=="QSUM")        {            scanf("%lld%lld",&x,&y);            printf("%lld\n",query(x,y));        }        else if(mode=="CHANGE")        {            scanf("%lld%lld",&x,&z);            change(1,tr[x].s,tr[x].s,z);        }        else if(mode=="QMAX")        {            scanf("%lld%lld",&x,&y);             printf("%lld\n",find(x,y));        }    }    return 0;}

注意这题输入有负数，求ans要搞好初值