【树链剖分】【分类讨论】水果姐逛水果街Ⅲ

3306 水果姐逛水果街Ⅲ

 时间限制: 2 s
 空间限制: 256000 KB
 题目等级 : 大师 Master

题目描述 Description

连续两天都没有被cgh难倒，水果姐心情很不错，第三天又来逛
水果街。
今天cgh早早就来到了水果街等水果姐，因为他带来了帮手cys。cgh的魔法是把水果街变成树结构，而cys的魔法是瞬间改变某家店的水果价格。一边问一边改，绝不给水果姐思考的时间！
同样还是n家水果店，编号为1~n，每家店能买水果也能卖水果，并且同一家店卖与买的价格一样。
cgh和cys一共有m个操作。
操作的格式为
0 x y 或 1 x y
如果操作类型是0，表示cys把第x家店的价格改为y
如果操作类型是1，则表示要求水果姐从第x家店出发到第y家店，途中只能选一家店买一个水果，然后选一家店（可以是同一家店，但不能往回走）卖出去。并且输出最多可以赚多少钱。
这题是给天牛做的。

输入描述 Input Description

第一行n，表示有n家店
下来n个正整数，表示每家店一个苹果的价格。
下来n-1行，每行两个整数x，y，表示第x家店和第y家店有一条边。
下来一个整数m，表示下来有m个操作。
下来有m行，每行三个整数，表示一次操作。

输出描述 Output Description

每行对应一个1类型的操作，输出一个整数，表示面对cgh的每次询问，水果姐最多可以赚到多少钱。

样例输入 Sample Input

4
16 5 1 15
1 2
1 3
2 4
3
1 3 4
0 3 17
1 4 3

样例输出 Sample Output

15
12

数据范围及提示 Data Size & Hint

0<=苹果的价格<=10^8
0<n<=2*10^5
0<m<=10^5

题解：

因为有修改，所以写了树链剖分。查询的时候和写Ⅱ的时候有异曲同工之妙。。一样恶心。.

Code：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#define N 200100
#define root 1,1,tot
#define lchild rt<<1,l,m
#define rchild rt<<1|1,m+1,r
using namespace std;
struct Edge{
    int v,next;
}edge[2*N];
struct Tree{
    int maxn,minn,cq,ch;
}tree[4*N];
int n,q,num=0,tot=0;
int head[N],fa[N],de[N],dv[N],sz[N],son[N],w[N],top[N];
bool vis[N]={0};
int in(){
    int x=0; char ch=getchar();
    while (ch<'0' || ch>'9') ch=getchar();
    while (ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}
void add(int u,int v){
    edge[++num].v=v; edge[num].next=head[u]; head[u]=num;
}
void dfs1(int x){
    sz[x]=1; vis[x]=true;
    for (int i=head[x]; i; i=edge[i].next){
        int v=edge[i].v;
        if (vis[v]) continue;
        fa[v]=x; de[v]=de[x]+1;
        dfs1(v); sz[x]+=sz[v];
        if (sz[son[x]]<sz[v]) son[x]=v;
    }
}
void dfs2(int x,int start){
    w[x]=++tot; top[x]=start;
    if (!son[x]) return; dfs2(son[x],start);
    for (int i=head[x]; i; i=edge[i].next)
        if (edge[i].v!=fa[x] && edge[i].v!=son[x])
            dfs2(edge[i].v,edge[i].v);
}
void push_up(int rt){
    tree[rt].maxn=max(tree[rt<<1].maxn,tree[rt<<1|1].maxn);
    tree[rt].minn=min(tree[rt<<1].minn,tree[rt<<1|1].minn);
    tree[rt].cq=max(max(tree[rt<<1].cq,tree[rt<<1|1].cq),tree[rt<<1|1].maxn-tree[rt<<1].minn);
    tree[rt].ch=max(max(tree[rt<<1].ch,tree[rt<<1|1].ch),tree[rt<<1].maxn-tree[rt<<1|1].minn);
}
void update(int rt,int l,int r,int x,int p){
    if (l==r){
        tree[rt].maxn=tree[rt].minn=p;
        tree[rt].cq=tree[rt].ch=0;
        return;
    }
    int m=(l+r)>>1;
    if (x<=m) update(lchild,x,p);
    else update(rchild,x,p);
    push_up(rt);
}
Tree find(int rt,int l,int r,int ll,int rr){
    if (ll<=l && r<=rr) return (Tree)tree[rt];
    int m=(l+r)>>1; Tree s1,s2,s3;
    s1=s2=s3=(Tree){0,0x7fffffff,0,0};
    if (ll<=m) s1=find(lchild,ll,rr);
    if (rr>m) s2=find(rchild,ll,rr);
    s3.maxn=max(s1.maxn,s2.maxn);
    s3.minn=min(s1.minn,s2.minn);
    s3.cq=max(max(s1.cq,s2.cq),s2.maxn-s1.minn);
    s3.ch=max(max(s1.ch,s2.ch),s1.maxn-s2.minn);
    return (Tree)s3;
}
void build(){
    for (int i=1; i<=n; i++)
        update(root,w[i],dv[i]);
}
int query(int x,int y){
    int f1=top[x],f2=top[y],sx=0,sy=0,maxx=0,maxy=0,minx=0x7fffffff,miny=0x7fffffff;
    Tree ss=(Tree){0,0x7fffffff,0,0};
    while (f1!=f2){
        if (de[f1]>de[f2]){
            ss=find(root,w[f1],w[x]);
            sx=max(sx,max(ss.ch,ss.maxn-minx));
            maxx=max(maxx,ss.maxn); minx=min(minx,ss.minn);
            x=fa[f1]; f1=top[x];
        }
        else {
            ss=find(root,w[f2],w[y]);
            sy=max(sy,max(ss.cq,maxy-ss.minn));
            maxy=max(maxy,ss.maxn); miny=min(miny,ss.minn);
            y=fa[f2]; f2=top[y];
        }
    }
    if (de[x]>de[y]){
        ss=find(root,w[y],w[x]);
        sx=max(sx,max(ss.ch,ss.maxn-minx));
        maxx=max(maxx,ss.maxn); minx=min(minx,ss.minn);
    }
    else {
        ss=find(root,w[x],w[y]);
        sy=max(sy,max(ss.cq,maxy-ss.minn));
        maxy=max(maxy,ss.maxn); miny=min(miny,ss.minn);
    }
    return max(max(sx,sy),maxy-minx);
}
int main(){
    n=in();
    for (int i=1; i<=n; i++) dv[i]=in();
    for (int i=1; i<n; i++){
        int u=in(),v=in();
        add(u,v),add(v,u);
    }
    dfs1(1); dfs2(1,1); build();
    q=in();
    while (q--){
        int opt=in(),x=in(),y=in();
        if (!opt) update(root,w[x],y);
        else printf("%d\n",query(x,y));
    }
    return 0;
}