bzoj3999 [TJOI2015]旅游

Description

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

Input

第一行输入一个正整数N，表示城市个数。

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

第三行开始连续输入N-1行，每行有两个数字x和y。表示x城市和y城市有一条路径。城市编号从1开始。

下一行输入一个整数Q，表示询问次数。

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

1≤ N≤50000， 1≤Q ≤50000

Output

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

Sample Input

3
1 2 3
1 2
2 3
2
1 2 100
1 3 100

Sample Output

1
1

 

正解：树链剖分+线段树。

这题还是有点代码难度的，不过想清楚再写就很容易了。。

我们考虑维护4个东西，最小值，最大值，前缀最大减后缀最小，后缀最大减前缀最小，注意线段树的叶子结点的后两个标记必须为$0$，不然会出错。

如何合并标记？前两个标记很好合并，但是后两个似乎有点麻烦。其实很简单，后两个标记的合并可以分为两部分，一部分是两个儿子的标记取最值，还有一部分是左儿子的最大值减右儿子的最小值或左儿子的最小值减右儿子的最大值，两部分都取最大值就行了。

这题区间修改很$simple$，就是最小值和最大值加上修改的$w$就行了。

考虑如何查询。我们在跳链时会扣出$logn$个区间，那么答案由两部分组成。

一部分是一个区间以内的，这个部分就是这个区间的前缀最大减后缀最小或后缀最大减前缀最小。考虑如果这个区间的右端点更靠近$a$点，那么答案就是前者，否则是后者。

另一部分是不同区间之间的。那么首先我们要对这个区间按照从$a$到$b$经过的顺序排序。所以我们肯定要对每一个区间加上标记，具体的话看看我的代码就知道了。

然后我们直接$O(log^{2}n)$地暴力枚举两两区间。易知答案就是后一个区间的最大值减去前一个区间的最小值。

对于两个部分的答案，我们直接取最大值输出就行了。

 

//It is made by wfj_2048~
#include <algorithm>
#include <iostream>
#include <complex>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <vector>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <set>
#define inf (1<<30)
#define N (100010)
#define ls (x<<1)
#define rs (x<<1|1)
#define il inline
#define RG register
#define LL long long
#define File(s) freopen(s".in","r",stdin),freopen(s".out","w",stdout)

using namespace std;

struct edge{ LL nt,to; }g[2*N];
struct data{ LL mn,mx,ll,rr; };
struct node{ LL cnt,t; data res; }q[110];

LL head[N],top[N],fa[N],son[N],p[N],sz[N],dep[N],tid[N],pos[N],n,num,cnt,Q;
LL lazy[4*N],mn[4*N],mx[4*N],ll[4*N],rr[4*N];

il LL gi(){
    RG LL x=0,q=1; RG char ch=getchar();
    while ((ch<'0' || ch>'9') && ch!='-') ch=getchar();
    if (ch=='-') q=-1,ch=getchar();
    while (ch>='0' && ch<='9') x=x*10+ch-48,ch=getchar();
    return q*x;
}

il void insert(RG LL from,RG LL to){
    g[++num]=(edge){head[from],to},head[from]=num; return;
}

il LL cmpq(const node &a,const node &b){
    if (a.cnt!=b.cnt) return a.cnt<b.cnt;
    if (a.cnt==1) return a.t<b.t; return a.t>b.t;
}

il void dfs1(RG LL x,RG LL p){
    fa[x]=p,dep[x]=dep[p]+1,sz[x]=1; RG LL v;
    for (RG LL i=head[x];i;i=g[i].nt){
    v=g[i].to; if (v==p) continue;
    dfs1(v,x); sz[x]+=sz[v];
    if (sz[son[x]]<=sz[v]) son[x]=v;
    }
    return;
}

il void dfs2(RG LL x,RG LL p,RG LL anc){
    top[x]=anc,tid[x]=++cnt,pos[cnt]=x;
    if (son[x]) dfs2(son[x],x,anc); RG LL v;
    for (RG LL i=head[x];i;i=g[i].nt){
    v=g[i].to; if (v==p || v==son[x]) continue;
    dfs2(v,x,v);
    }
    return;
}

il void pushdown(RG LL x){
    mn[ls]+=lazy[x],mn[rs]+=lazy[x];
    mx[ls]+=lazy[x],mx[rs]+=lazy[x];
    lazy[ls]+=lazy[x],lazy[rs]+=lazy[x];
    lazy[x]=0; return;
}

il void merge(RG LL x,RG LL l,RG LL r){
    ll[x]=max(ll[ls],ll[rs]),ll[x]=max(ll[x],mx[ls]-mn[rs]);
    rr[x]=max(rr[ls],rr[rs]),rr[x]=max(rr[x],mx[rs]-mn[ls]);
    mn[x]=min(mn[ls],mn[rs]),mx[x]=max(mx[ls],mx[rs]); return;
}

il void build(RG LL x,RG LL l,RG LL r){
    if (l==r){ mn[x]=mx[x]=p[pos[l]]; return; }
    RG LL mid=(l+r)>>1; build(ls,l,mid),build(rs,mid+1,r);
    merge(x,l,r); return;
}

il void update(RG LL x,RG LL l,RG LL r,RG LL xl,RG LL xr,RG LL v){
    if (xl<=l && r<=xr){ mn[x]+=v,mx[x]+=v,lazy[x]+=v; return; }
    if (lazy[x]) pushdown(x); RG LL mid=(l+r)>>1;
    if (xr<=mid) update(ls,l,mid,xl,xr,v);
    else if (xl>mid) update(rs,mid+1,r,xl,xr,v);
    else update(ls,l,mid,xl,mid,v),update(rs,mid+1,r,mid+1,xr,v);
    merge(x,l,r); return;
}

il data query(RG LL x,RG LL l,RG LL r,RG LL xl,RG LL xr){
    if (xl<=l && r<=xr) return (data){mn[x],mx[x],ll[x],rr[x]};
    if (lazy[x]) pushdown(x); RG LL mid=(l+r)>>1;
    if (xr<=mid) return query(ls,l,mid,xl,xr);
    else if (xl>mid) return query(rs,mid+1,r,xl,xr);
    else{
    RG data res1=query(ls,l,mid,xl,mid);
    RG data res2=query(rs,mid+1,r,mid+1,xr);
    RG LL ll=max(res1.ll,res2.ll); ll=max(ll,res1.mx-res2.mn);
    RG LL rr=max(res1.rr,res2.rr); rr=max(rr,res2.mx-res1.mn);
    RG LL mn=min(res1.mn,res2.mn),mx=max(res1.mx,res2.mx);
    return (data){mn,mx,ll,rr};
    }
}

il void change(RG LL u,RG LL v,RG LL w){
    while (top[u]!=top[v]){
    if (dep[top[u]]<dep[top[v]]) swap(u,v);
    update(1,1,n,tid[top[u]],tid[u],w),u=fa[top[u]];
    }
    if (dep[u]>dep[v]) swap(u,v);
    update(1,1,n,tid[u],tid[v],w); return;
}

il LL Query(RG LL u,RG LL v){
    RG LL uu=0,vv=0,cntu=1,cntv=2,cnt=0,res=0;
    while (top[u]!=top[v]){
    if (dep[top[u]]<dep[top[v]])
        swap(u,v),swap(uu,vv),swap(cntu,cntv);
    q[++cnt]=(node){cntu,++uu,query(1,1,n,tid[top[u]],tid[u])};
    u=fa[top[u]];
    }
    if (dep[u]>dep[v]) swap(u,v),swap(uu,vv),swap(cntu,cntv);
    q[++cnt]=(node){cntv,++vv,query(1,1,n,tid[u],tid[v])};
    sort(q+1,q+cnt+1,cmpq);
    for (RG LL i=1;i<=cnt;++i)
    if (q[i].cnt==1) res=max(res,q[i].res.ll);
    else res=max(res,q[i].res.rr);
    for (RG LL i=1;i<cnt;++i)
    for (RG LL j=i+1;j<=cnt;++j)
        res=max(res,q[j].res.mx-q[i].res.mn);
    return res;
}

il void work(){
    n=gi();
    for (RG LL i=1;i<=n;++i) p[i]=gi();
    for (RG LL i=1,u,v;i<n;++i)
    u=gi(),v=gi(),insert(u,v),insert(v,u);
    dfs1(1,0),dfs2(1,0,1),build(1,1,n),Q=gi();
    for (RG LL i=1,u,v,w;i<=Q;++i){
    u=gi(),v=gi(),w=gi();
    printf("%lld\n",Query(u,v));
    change(u,v,w);
    }
    return;
}

int main(){
    File("travel");
    work();
    return 0;
}