洛谷 P3976 [TJOI2015]旅游 树链剖分

题目描述

为了提高智商,ZJY准备去往一个新世界去旅游。这个世界的城市布局像一棵树。每两座城市之间只有一条路径可以互达。每座城市都有一种宝石,有一定的价格。ZJY为了赚取最高利益,她会选择从A城市买入再转手卖到B城市。由于ZJY买宝石时经常卖萌,因而凡是ZJY路过的城市,这座城市的宝石价格会上涨。让我们来算算ZJY旅游完之后能够赚取的最大利润。(如a城市宝石价格为v,则ZJY出售价格也为v)

输入输出格式

输入格式：
第一行输入一个正整数N表示城市个数

接下来一行输入N个正整数表示每座城市宝石的最初价格p,每个宝石的初始价格不超过100。

第三行开始连续输入N-1行,每行有两个数字x和y。表示x城市和y城市有一条路径。城市编号从1开始。下一行输入一个正整数Q表示询问次数。

接下来Q行每行输入三个正整数a,b,U,表示ZY从a旅游到b,城市宝石上涨v。

输出格式：
对于每次询问,输出ZJY可能获得的最大利润,如果亏本了则输出0。

输入输出样例

输入样例#1：
3
1 2 3
1 2
2 3
2
1 2 100
1 3 100
输出样例#1：
1
1
说明

数据范围

对于 $30\%$ 的数据，有 $0<N\le 100,0<Q\le 10000$。

对于 $100\%$ 的数据，有 $0<N\le 50000,0<Q\le 50000$。

分析：
对于一个问题相当于求路径上前面最大减后面最小值。
我们可以线段树维护一个$min$和一个$max$表示当前区间最大最小值。然后维护一个$pre$和$next$表示前面小后面大的权值和前面大后面小的权值。
因为对于一条链，左边的链的方向与dfs序相反，右边的链与dfs相同。那么左边就要取next，右边取pre。合并两条链时可以先交换pre和next再搞。

代码：

// luogu-judger-enable-o2
#include <iostream>
#include <cstdio>
#include <cmath>

const int maxn=5e4+7;

using namespace std;

int dfn[maxn],top[maxn],a[maxn],ls[maxn],size[maxn],fa[maxn],dep[maxn],lazy[maxn*4];
int n,m,x,y,w,cnt;

struct edge{
    int y,next;
}g[maxn*2];

struct node{
    int pre,next,maxx,minn;
}t[maxn*4];

void add(int x,int y)
{
    g[++cnt]=(edge){y,ls[x]};
    ls[x]=cnt;
}

void dfs1(int x,int f)
{
    size[x]=1;
    fa[x]=f;
    dep[x]=dep[f]+1;
    for (int i=ls[x];i>0;i=g[i].next)
    {
        int y=g[i].y;
        if (y==fa[x]) continue;
        dfs1(y,x);
        size[x]+=size[y];
    }
}

void dfs2(int x,int f)
{
    top[x]=f;
    dfn[x]=++cnt;
    int c=0;
    for (int i=ls[x];i>0;i=g[i].next)
    {
        int y=g[i].y;
        if (y==fa[x]) continue;
        if (size[c]<size[y]) c=y;
    }
    if (!c) return;
    dfs2(c,f);
    for (int i=ls[x];i>0;i=g[i].next)
    {
        int y=g[i].y;
        if ((y==fa[x]) || (y==c)) continue;
        dfs2(y,y);
    }
}

node merge(node x,node y)
{
    node z;
    z.maxx=max(x.maxx,y.maxx);
    z.minn=min(x.minn,y.minn);
    z.pre=max(max(x.pre,y.pre),y.maxx-x.minn);
    z.next=max(max(x.next,y.next),x.maxx-y.minn);
    return z;
}

void clean(int p)
{
    if (lazy[p])
    {
        t[p*2].maxx+=lazy[p];
        t[p*2].minn+=lazy[p];
        t[p*2+1].maxx+=lazy[p];
        t[p*2+1].minn+=lazy[p];
        lazy[p*2]+=lazy[p];
        lazy[p*2+1]+=lazy[p];
        lazy[p]=0;
    }
}

void ins(int p,int l,int r,int x,int y,int k)
{
    if ((l==x) && (r==y))
    {
        lazy[p]+=k;
        t[p].maxx+=k;
        t[p].minn+=k;
        return;
    }
    clean(p);
    int mid=(l+r)/2;
    if (y<=mid) ins(p*2,l,mid,x,y,k);
    else if (x>mid) ins(p*2+1,mid+1,r,x,y,k);
    else
    {
        ins(p*2,l,mid,x,mid,k);
        ins(p*2+1,mid+1,r,mid+1,y,k);
    }
    t[p]=merge(t[p*2],t[p*2+1]);
}

node query(int p,int l,int r,int x,int y)
{
    if ((l==x) && (r==y)) return t[p];
    clean(p);
    int mid=(l+r)/2;
    if (y<=mid) return query(p*2,l,mid,x,y);
    else if (x>mid) return query(p*2+1,mid+1,r,x,y);
    else return merge(query(p*2,l,mid,x,mid),query(p*2+1,mid+1,r,mid+1,y));
}

void change(int x,int y,int k)
{
    while (top[x]!=top[y])
    {
        if (dep[top[x]]>dep[top[y]]) swap(x,y);
        ins(1,1,n,dfn[top[y]],dfn[y],k);
        y=fa[top[y]];
    }
    if (dep[x]>dep[y]) swap(x,y);
    ins(1,1,n,dfn[x],dfn[y],k);
}

int getans(int x,int y)
{
    int flagx=1,flagy=0;
    node dx=(node){0,0,0,0x3f3f3f3f},dy=(node){0,0,0,0x3f3f3f3f};
    while (top[x]!=top[y])
    {
        if (dep[top[x]]>dep[top[y]]) swap(x,y),swap(dx,dy),swap(flagx,flagy);
        node d=query(1,1,n,dfn[top[y]],dfn[y]);
        if (flagy) swap(d.pre,d.next),dy=merge(dy,d);
              else dy=merge(d,dy);
        y=fa[top[y]];
    }
    if (dep[x]>dep[y]) swap(x,y),swap(dx,dy),swap(flagx,flagy);
    node d=query(1,1,n,dfn[x],dfn[y]);
    if (flagy) swap(d.pre,d.next),dy=merge(dy,d);
          else dy=merge(d,dy);
    if (!flagx) swap(x,y),swap(dx,dy),swap(flagx,flagy);
    dx=merge(dx,dy);
    return dx.pre;
}

int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    for (int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    dfs1(1,0);
    cnt=0;
    dfs2(1,0);	
    for (int i=1;i<=n;i++) ins(1,1,n,dfn[i],dfn[i],a[i]);		
    scanf("%d",&m); 
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&w);
        printf("%d\n",getans(x,y));
        change(x,y,w);
    }
}