gym101138J(树链剖分,线段树维护区间连续子段最大和,好题)

J. Valentina and the Gift Tree
time limit per test
6 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output

The Valentina's birthday is coming soon! Her parents love her so much that they are preparing some gifts to give her. Valentina is cute and smart and her parents are planning to set an interesting task for her.

They prepared a tree (a connected graph without cycles) with one gift in each vertex. Vertices are numbered 1 through n and the i-th of them contains a gift with value gi (a value describing Valentina's happiness if she gets a gift from this vertex). All gifts are wrapped so Valentina doesn't know their values.

Note that gi may be negative and it would mean that Valentina doesn't like the gift (do you remember when your parents gave you clothes or toys not adequate to your interests?).

Let's consider the following process:

  1. Valentina chooses two vertices a and b (not necessarily distinct). 
  2. She unpacks all gifts on the path connecting a and b and thus she sees their values. 
  3. She chooses some part of the path connecting a and b. The chosen part must be connected and can't be empty (hence, it must be a path). 
  4. Valentina takes all gifts in the chosen part of the path and her total happiness is equal to the sum of the values of taken gifts. 

Valentina is smart and for chosen a and b she will choose a part resulting in the biggest total happiness.

In order to maximize her chance to get a good bunch of gifts, parents allow her to ask q questions, each with two numbers a and b. For each Valentina's question parents will tell her the answer, i.e. the maximum total happiness for chosen a and b.

They noticed that answering Valentina's questions is an interesting problem. They want to know if you are smart enough to correctly answer all questions.

Input

The first line of the input contains one integer n (2 ≤ n ≤ 105) — the size of the tree.

Each of the next n - 1 lines contains two integers ui and vi (1 ≤ ui, vi ≤ n) — indices of two vertices connected with an edge. The given edges are guaranteed to describe a tree.

The next line contains n integers g1, g2, ... gn ( - 109 ≤ gi ≤ 109).

Then, the questions are given. One line contains an integer q (1 ≤ q ≤ 500 000) — the number of questions.

Finally, each of the last q lines contains two integers ai and bi (1 ≤ ai, bi ≤ n), describing one question.

Output

For each of the q questions find the maximum happiness of Valentina if she has to choose a non-empty part of the given path. For every question print the answer in a separate line.

Examples
input
6
1 2
1 3
1 4
4 5
4 6
-1 2 2 -2 5 8
6
2 5
2 3
1 5
5 3
1 4
5 6
output
5
3
5
5
-1
11

题意:

给出n个点的一棵树,每个结点有个权值,m个查询,每个查询给出两个点u和v,查询u到v路径上最大连续和。


题解:

先树链剖分,在线段树中维护最大连续子段和,那么合并的时候还需要记录区间总和,区间左部分连续最大和,区间右部分最大连续和。

与以往不同的是,查询操作时返回一个线段树的结点类型的数,以便在树链剖分查询时的合并和线段树查询操作的合并。

详细见代码:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<queue>
#include<stack>
using namespace std;
#define rep(i,a,n) for (int i=a;i<n;i++)
#define per(i,a,n) for (int i=n-1;i>=a;i--)
#define pb push_back
#define fi first
#define se second
typedef vector<int> VI;
typedef long long ll;
typedef pair<int,int> PII;
const ll inf=0x3ffffffffffff;
const ll mod=1000000007;
const int maxn=100000+100;
int head[maxn];
struct edge
{
    int from,to,next;
}e[maxn*2];   //
struct node
{
    int l,r;
    ll mx,sum,lm,rm;
    node(){
        mx=sum=lm=rm=-inf;
        return;
    }
}seg[maxn*4];
int tol=0;
void add(int u,int v)
{
    e[++tol].to=v,e[tol].next=head[u],head[u]=tol;
}

int pos;
int fa[maxn],top[maxn],son[maxn],num[maxn],p[maxn],dep[maxn],fp[maxn];
ll a[maxn];
node up(node ls,node rs)
{
    node t;
    t.l=ls.l,t.r=rs.r;
    t.mx=max(max(ls.mx,rs.mx),ls.rm+rs.lm);
    t.lm=max(ls.lm,ls.sum+rs.lm);
    t.rm=max(rs.rm,ls.rm+rs.sum);
    t.sum=ls.sum+rs.sum;
    return t;
}
void build(int i,int l,int r)
{
    seg[i].l=l,seg[i].r=r;
    if(l==r)
    {
        seg[i].sum=seg[i].mx=seg[i].lm=seg[i].rm=a[fp[l]];
        return;
    }
    int m=(l+r)/2;
    build(i*2,l,m),build(i*2+1,m+1,r);
    seg[i]=up(seg[i*2],seg[i*2+1]);
}
void dfs(int u,int f,int d)
{
    num[u]=1;
    fa[u]=f;
    dep[u]=d;
    for(int i=head[u];i;i=e[i].next)
    {
        int v=e[i].to;
        if(v==f) continue;
        dfs(v,u,d+1);
        if(son[u]==0||num[v]>num[son[u]])
            son[u]=v;
        num[u]+=num[v];
    }
}
void dfs2(int u,int sp)
{
    top[u]=sp;
    if(son[u])
    {
        p[u]=pos;
        fp[pos]=u;
        pos++;
        dfs2(son[u],sp);
    }
    else
    {
        p[u]=pos;
        fp[pos]=u;
        pos++;
        return;
    }
    for(int i=head[u];i;i=e[i].next)
    {
        int v=e[i].to;
        if(v!=son[u]&&v!=fa[u])
            dfs2(v,v);
    }
}

node query(int i,int l,int r)
{
    if(seg[i].l==l&&seg[i].r==r)
    {
        return seg[i];
    }
    int m=(seg[i].l+seg[i].r)/2;
    if(r<=m) return query(i*2,l,r);
    else if(l>m) return query(i*2+1,l,r);
    node t1=query(i*2,l,m),t2=query(i*2+1,m+1,r);
    return up(t1,t2);
}
ll find(int u,int v)
{
    int fu=top[u],fv=top[v];
    node tu,tv;         //
    bool f1=false,f2=false;
    while(fu!=fv)
    {
        if(dep[fu]>dep[fv])
        {
            if(f1)
            tu=up(query(1,p[fu],p[u]),tu);  //特别注意先后顺序,tu写在后面,后面的几个合并操作也是一样
            else tu=query(1,p[fu],p[u]),f1=true;
            u=fa[fu],fu=top[u];
        }
        else
        {
            if(f2) tv=up(query(1,p[fv],p[v]),tv);//
            else tv=query(1,p[fv],p[v]),f2=true;
            v=fa[fv],fv=top[v];
        }
    }
    if(dep[u]>dep[v])
    {
        if(f1)
        tu=up(query(1,p[v],p[u]),tu);
        else tu=query(1,p[v],p[u]);
    }
    else
    {
        if(f2)
            tv=up(query(1,p[u],p[v]),tv);
        else tv=query(1,p[u],p[v]);
    }
    swap(tu.lm,tu.rm);
    return up(tu,tv).mx;
}
void init()
{
    tol=0;pos=1;
    memset(head,0,sizeof(head));
    memset(fa,0,sizeof(fa));
    memset(son,0,sizeof(son));
}

int main()
{
    init();
    int n;
    scanf("%d",&n);
    rep(i,1,n)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v),add(v,u);
    }
    rep(i,1,n+1)
        scanf("%lld",&a[i]);
    dfs(1,0,0);
    dfs2(1,1);
    build(1,1,pos);
    int m;
    scanf("%d",&m);
    while(m--)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        printf("%lld\n",find(u,v));
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值