百度科技園內有nn個零食機，零食機之間通過n?1n?1條路相互連通。每個零食機都有一個值vv，表示為小度熊提供零食的價值。

由于零食被頻繁的消耗和補充，零食機的價值vv會時常發生變化。小度熊只能從編號為0的零食機出發，并且每個零食機至多經過一次。另外，小度熊會對某個零食機的零食有所偏愛，要求路線上必須有那個零食機。

為小度熊規劃一個路線，使得路線上的價值總和最大。

Input輸入數據第一行是一個整數T(T≤10)T(T≤10)，表示有TT組測試數據。

對于每組數據，包含兩個整數n,m(1≤n,m≤100000)n,m(1≤n,m≤100000)，表示有nn個零食機，mm次操作。

接下來n?1n?1行，每行兩個整數xx和y(0≤x,y<n)y(0≤x,y<n)，表示編號為xx的零食機與編號為yy的零食機相連。

接下來一行由nn個數組成，表示從編號為0到編號為n?1n?1的零食機的初始價值v(|v|<100000)v(|v|<100000)。

接下來mm行，有兩種操作：0xy0xy，表示編號為xx的零食機的價值變為yy；1x1x，表示詢問從編號為0的零食機出發，必須經過編號為xx零食機的路線中，價值總和的最大值。

本題可能棧溢出，辛苦同學們提交語言選擇c++，并在代碼的第一行加上：

`#pragma comment(linker, "/STACK:1024000000,1024000000") `Output對于每組數據，首先輸出一行”Case #?:”，在問號處應填入當前數據的組數，組數從1開始計算。

對于每次詢問，輸出從編號為0的零食機出發，必須經過編號為xx零食機的路線中，價值總和的最大值。
Sample Input

1
6 5
0 1
1 2
0 3
3 4
5 3
7 -5 100 20 -5 -7
1 1
1 3
0 2 -1
1 1
1 5

Sample Output

Case #1:
102
27
2
20

利用DFS序可以將對子樹的操作轉化為對區間的操作，線段樹

#pragma comment(linker, "/STACK:1024000000,1024000000") 
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<vector>
#include<string>
#include<cstring>
using namespace std;
typedef long long LL;
const int MAXN = 100002;
//一開始全部Insert一遍然后再操作
struct node
{
    LL l,r;
    LL max;//子樹中最大的節點
    LL laz;
}T[MAXN*4+9];
struct edge
{
    LL v,next;
}E[MAXN];
LL head[MAXN], cnt, index;
LL b[MAXN],e[MAXN], val[MAXN];
LL range[MAXN];
LL n, m;
void init()
{
    index = cnt = 0;
    memset(head,-1,sizeof(head));
}
void addedge(int u,int v)
{
    E[cnt].v = v;
    E[cnt].next = head[u];
    head[u] = cnt++;
}
void dfs(int u,int pre)
{
    b[u] = index++;
    range[u] = index - 1;
    for(int i = head[u];i!=-1;i = E[i].next)
    {
        if(E[i].v==pre) continue;
        dfs(E[i].v,u);
    }
    e[u] = index;
}
void pushdown(LL x)
{
    if(T[x].laz)
    {
        T[x*2].laz += T[x].laz;
        T[x*2+1].laz += T[x].laz;
        T[x*2].max += T[x].laz;
        T[x*2+1].max += T[x].laz;
        T[x].laz = 0;
    }
}
void pushup(LL x)
{
    T[x].max = max(T[x*2].max,T[x*2+1].max);
}
void build(LL x,LL l,LL r)
{
    T[x].l = l ,T[x].r = r,T[x].laz = 0, T[x].max = 0;
    if(l==r) return;
    LL mid = (l + r)/2;
    build(x*2,l,mid);
    build(x*2+1,mid+1,r);
    pushup(x);
}
void update(LL x,LL l,LL r,LL val)
{
    if(T[x].l == l && T[x].r == r)
    {
        T[x].max += val;
        T[x].laz += val;
        return ;
    }
    pushdown(x);
    LL mid = (T[x].l + T[x].r)/2;
    if(r<=mid)
        update(x*2,l,r,val);
    else if(l>mid)
        update(x*2+1,l,r,val);
    else
    {
        update(x*2,l,mid,val);
        update(x*2+1,mid+1,r,val);
    }
    pushup(x);
}
LL query(LL x, LL l, LL r)
{
    if(T[x].l == l && T[x].r == r)
    {
        return T[x].max;
    }
    pushdown(x);
    LL mid = ( T[x].l + T[x].r)/2;
    if(r<=mid)
        return query(x*2,l,r);
    else if(l>mid)
        return query(x*2+1,l,r);
    else 
        return max(query(x*2,l,mid),query(x*2+1,mid+1,r));
    
}
int main()
{
    int cas = 1;
    LL tt;
    scanf("%lld",&tt);
    while(tt--)
    {
        init();
        LL op,f,t;
        scanf("%lld%lld",&n,&m);
        for(LL i= 0;i<n - 1;i++)
        {
            scanf("%lld%lld",&f,&t);
            addedge(f,t);
            addedge(t,f);
        }
        dfs(0,-1);
        for(int i = 0;i<n;i++)
            scanf("%lld",&val[i]);
        build(1,0,n-1);
        for(int i= 0;i<n;i++)
            update(1,b[i],e[i]-1,val[i]);
        printf("Case #%d:
",cas++);
        while(m--)
        {
            scanf("%lld",&op);
            if(op==0)
            {
                scanf("%lld%lld",&f,&t);//把f點的價值修改為t
                update(1,b[f],e[f]-1,t-val[f]);
                val[f] = t;
            }
            else if(op==1)
            {
                scanf("%lld",&f);//查詢x點子樹最大值
                printf("%lld
",query(1,b[f],e[f]-1));
            }
        }
    }
}

總結

以上是生活随笔為你收集整理的Snacks的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。