HDU_5692_dfs序+线段树

Posted 2020-08-12 冷暖知不知

http://acm.hdu.edu.cn/showproblem.php?pid=5692

 

这道题真的是看了题解还搞了一天，把每条路径后序遍历按1-n重新标号，储存每个点在哪些路径中出现过（l和r数组），然后转化成线段树来更新和取最大值。

注意，如果使用递归建线段树，数组要开4n才能保证不超。

刚开始更新的函数每一遍更新了所有子树，然后超时了，后来在tree中加了个add，保存子树需要增加的量，如果用到这个子树，在把这个量加起来，简直巧妙。

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define INF 0x3f3f3f3f
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;

vector<int> line[100005];
int n,m,cnt,l[100005],r[100005];
long long w[100005],init[100005];
struct segtree
{
    int left,right;
    long long maxx,add;
}tree[400005];

void dfs(int now,int pre,long long sum)
{
    sum += w[now];
    int flag = 1;
    l[now] = INF;
    for(int i = 0;i < line[now].size();i++)
    {
        int next = line[now][i];
        if(next == pre) continue;
        flag = 0;
        dfs(next,now,sum);
        l[now] = min(l[now],l[next]);
    }
    init[cnt] = sum;
    r[now] = cnt++;
    if(flag)    l[now] = r[now];
}

void build(int pos,int l,int r)
{
    tree[pos].left = l;
    tree[pos].right = r;
    tree[pos].add = 0;
    if(l == r)  tree[pos].maxx = init[l];
    else
    {
        int mid = (l+r)/2;
        build(pos*2,l,mid);
        build(pos*2+1,mid+1,r);
        tree[pos].maxx = max(tree[pos*2].maxx,tree[pos*2+1].maxx);
    }
}

void update(int pos,int l,int r,long long v)
{
    if(tree[pos].add != 0)
    {
        if(tree[pos].left != tree[pos].right)
        {
            tree[pos*2].maxx += tree[pos].add;
            tree[pos*2].add += tree[pos].add;
            tree[pos*2+1].maxx += tree[pos].add;
            tree[pos*2+1].add += tree[pos].add;
            tree[pos].add = 0;
        }
    }
    if(tree[pos].left == 
       l && r == tree[pos].right)
    {
        tree[pos].maxx += v;
        tree[pos].add += v;
        return;
    }
    int mid = (tree[pos].left+tree[pos].right)/2;
    if(r <= mid)    update(pos*2,l,r,v);
    else if(l > mid)    update(pos*2+1,l,r,v);
    else
    {
        update(pos*2,l,mid,v);
        update(pos*2+1,mid+1,r,v);
    }
    tree[pos].maxx = max(tree[pos*2].maxx,tree[pos*2+1].maxx);
}

long long getmax(int pos,int l,int r)
{
    if(tree[pos].add != 0)
    {
        if(tree[pos].left != tree[pos].right)
        {
            tree[pos*2].maxx += tree[pos].add;
            tree[pos*2].add += tree[pos].add;
            tree[pos*2+1].maxx += tree[pos].add;
            tree[pos*2+1].add += tree[pos].add;
            tree[pos].add = 0;
        }
    }
    if(tree[pos].left == l && r == tree[pos].right) return tree[pos].maxx;
    int mid = (tree[pos].left+tree[pos].right)/2;
    if(r <= mid)    return getmax(pos*2,l,r);
    if(l > mid)     return getmax(pos*2+1,l,r);
    return max(getmax(pos*2,l,mid),getmax(pos*2+1,mid+1,r));
}

int main()
{
    int T;
    scanf("%d",&T);
    for(int z = 1;z <= T;z++)
    {
        printf("Case #%d:\n",z);
        scanf("%d%d",&n,&m);
        for(int i = 0;i < n;i++)    line[i].clear();
        for(int i = 0;i < n-1;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            line[x].push_back(y);
            line[y].push_back(x);
        }
        for(int i = 0;i < n;i++)    scanf("%lld",&w[i]);
        cnt = 1;
        dfs(0,0,0);
        build(1,1,n);
        while(m--)
        {
            int op;
            scanf("%d",&op);
            if(op == 0)
            {
                int x,y;
                scanf("%d%d",&x,&y);
                long long temp = y-w[x];
                w[x] = y;
                update(1,l[x],r[x],temp);
            }
            else
            {
                int x;
                scanf("%d",&x);
                printf("%lld\n",getmax(1,l[x],r[x]));
            }
        }
    }
    return 0;
}