bzoj 3999: [TJOI2015]旅游 树链剖分+线段树

题意

给一棵树，有点权。每次一个操作x y z表示询问从x出发走到y的路径上最大的a[i]-a[j]（必须满足先到i再到j）。然后把路径上的点权加上z。
n,q<=50000

分析

一开始看的时候没啥思路，想了想发现如果用线段树维护的话是可以合并区间的。
那么我们只要用线段树维护两个值，一个是正着走的答案，一个是反着走的答案。查询的时候把路径拆成两条分别查询即可。

代码

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=50005;
const LL inf=1e15;

int n,cnt,last[N],pos[N],dep[N],fa[N],top[N],size[N],sz,a[N],bel[N];
struct tree{LL mn,mx,ans1,ans2,tag;}t[N*5];
struct edge{int to,next;}e[N*2];

void addedge(int u,int v)
{
    e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].next=last[v];last[v]=cnt;
}

void dfs1(int x)
{
    size[x]=1;dep[x]=dep[fa[x]]+1;
    for (int i=last[x];i;i=e[i].next)
    {
        if (e[i].to==fa[x]) continue;
        fa[e[i].to]=x;
        dfs1(e[i].to);
        size[x]+=size[e[i].to];
    }
}

void dfs2(int x,int chain)
{
    pos[x]=++sz;bel[sz]=x;top[x]=chain;int k=0;
    for (int i=last[x];i;i=e[i].next)
        if (e[i].to!=fa[x]&&size[e[i].to]>size[k]) k=e[i].to;
    if (!k) return;
    dfs2(k,chain);
    for (int i=last[x];i;i=e[i].next)
        if (e[i].to!=fa[x]&&e[i].to!=k) dfs2(e[i].to,e[i].to);
}

int get_lca(int x,int y)
{
    while (top[x]!=top[y])
    {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        x=fa[top[x]];
    }
    if (dep[x]<dep[y]) return x;
    else return y;
}

tree merge(tree a,tree b)
{
    tree c;c.tag=0;
    c.ans1=max(a.ans1,max(b.ans1,a.mx-b.mn));
    c.ans2=max(a.ans2,max(b.ans2,b.mx-a.mn));
    c.mx=max(a.mx,b.mx);c.mn=min(a.mn,b.mn);
    return c;
}

void pushdown(int d,int l,int r)
{
    if (l==r||!t[d].tag) return;
    LL w=t[d].tag;t[d].tag=0;
    t[d*2].mn+=w;t[d*2].mx+=w;t[d*2].tag+=w;
    t[d*2+1].mn+=w;t[d*2+1].mx+=w;t[d*2+1].tag+=w;
}

void build(int d,int l,int r)
{
    if (l==r)
    {
        t[d].mn=t[d].mx=a[bel[l]];t[d].ans1=t[d].ans2=-inf;
        return;
    }
    int mid=(l+r)/2;
    build(d*2,l,mid);build(d*2+1,mid+1,r);
    t[d]=merge(t[d*2],t[d*2+1]);
}

tree Sig_query(int d,int l,int r,int x,int y)
{
    pushdown(d,l,r);
    if (l==x&&r==y) return t[d];
    int mid=(l+r)/2;
    if (y<=mid) return Sig_query(d*2,l,mid,x,y);
    else if (x>mid) return Sig_query(d*2+1,mid+1,r,x,y);
    else
    {
        tree u=Sig_query(d*2,l,mid,x,mid),v=Sig_query(d*2+1,mid+1,r,mid+1,y);
        return merge(u,v);
    }
}

LL query(int x,int y)
{
    int lca=get_lca(x,y);
    tree u,v;v.mn=inf;v.mx=-inf;v.ans1=v.ans2=-inf;
    if (x!=lca)
    {
        while (top[x]!=top[lca])
        {
            u=Sig_query(1,1,n,pos[top[x]],pos[x]);
            x=fa[top[x]];v=merge(u,v);
        }
        if (dep[x]<dep[lca]) u=Sig_query(1,1,n,pos[x],pos[lca]);
        else u=Sig_query(1,1,n,pos[lca],pos[x]);
        v=merge(u,v);
    }
    tree w=v;v.mn=inf;v.mx=-inf;v.ans1=v.ans2=-inf;
    if (y!=lca)
    {
        while (top[y]!=top[lca])
        {
            u=Sig_query(1,1,n,pos[top[y]],pos[y]);
            y=fa[top[y]];v=merge(u,v);
        }
        if (dep[y]<dep[lca]) u=Sig_query(1,1,n,pos[y],pos[lca]);
        else u=Sig_query(1,1,n,pos[lca],pos[y]);
        v=merge(u,v);
    }
    return max(v.ans2,max(w.ans1,v.mx-w.mn));
}

void Sig_ins(int d,int l,int r,int x,int y,int z)
{
    if (x>y) return;
    pushdown(d,l,r);
    if (l==x&&r==y)
    {
        t[d].mn+=z;t[d].mx+=z;t[d].tag+=z;
        return;
    }
    int mid=(l+r)/2;
    Sig_ins(d*2,l,mid,x,min(y,mid),z);
    Sig_ins(d*2+1,mid+1,r,max(mid+1,x),y,z);
    t[d]=merge(t[d*2],t[d*2+1]);
}

void ins(int x,int y,int z)
{
    while (top[x]!=top[y])
    {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        Sig_ins(1,1,n,pos[top[x]],pos[x],z);
        x=fa[top[x]];
    }
    if (dep[x]<dep[y]) Sig_ins(1,1,n,pos[x],pos[y],z);
    else Sig_ins(1,1,n,pos[y],pos[x],z);
}

int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    for (int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        addedge(x,y);
    }
    dfs1(1);
    dfs2(1,1);
    build(1,1,n);
    int T;
    scanf("%d",&T);
    while (T--)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        printf("%lld\n",max(query(x,y),(LL)0));
        ins(x,y,z);
    }
    return 0;
}