(线段树+dfs序)

来源:互联网 发布:js正则判断正整数 编辑:程序博客网 时间:2024/06/04 20:02

百度科技园内有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行,有两种操作:0 x y0 x y,表示编号为xx的零食机的价值变为yy;1 x1 x,表示询问从编号为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序:把一棵树的前序遍历转化为区间来进行操作,分别存储一个节点子树的左端点和右端点(好像叫做时间戳)然后就能用线段树来进行查询了,在本题中,线段树存储的是从根节点出发,到现在节点经过的总价值,必须要经过某点,就是要求以该节点为根的子树节点中到0的最大值
    dfs序
#pragma comment(linker, "/STACK:1024000000,1024000000") `#include <iostream>#include <algorithm>#include <cmath>#include <cstdio>#include <cstring>#include <vector>using namespace std;typedef long long ll;const int maxn = 100010;ll segtree[maxn*20];ll add[maxn*20];ll b[maxn];ll dis[maxn];int cnt=0,pa=0;int head[maxn];int n,m;void init(){    cnt=0;    pa=0;    memset(head,-1,sizeof(head));}struct Edge{    int to;    int next;}e[2*maxn];void adde(int a,int b){    e[pa].next=head[a];    e[pa].to=b;    head[a]=pa++;}struct Node{    int l,r;    ll v;}node[maxn*2];void  dfs(int x,int fx){    cnt++;    b[cnt]=dis[x];    node[x].l=cnt;    for(int i=head[x];i!=-1;i=e[i].next)    {        int cur=e[i].to;        if(cur==fx)continue;        dis[cur]=dis[x]+node[cur].v;        dfs(cur,x);    }    node[x].r=cnt;}void PushDown(ll rt,ll l){    if(add[rt]!=0)    {        add[rt<<1]+=add[rt];        add[rt<<1|1]+=add[rt];        segtree[rt<<1]+=add[rt];        segtree[rt<<1|1]+=add[rt];        add[rt]=0;    }}void plant(ll left,ll right,ll rt){    add[rt]=0;    if(left==right)    {        segtree[rt]=b[left];        return;    }    ll mid=(left+right)>>1;    plant(left,mid,rt<<1);    plant(mid+1,right,rt<<1|1);    segtree[rt]=max(segtree[rt<<1],segtree[rt<<1|1]);}ll ans;void query(ll left,ll right,ll rt,ll L,ll R){    if(L<=left&&right<=R)    {        if(ans<segtree[rt])ans=segtree[rt];        return;    }    PushDown(rt,right-left+1);    ll mid=(left+right)/2;    if(R<=mid)        query(left,mid,rt<<1,L,R);    else if(L>mid)        query(mid+1,right,rt<<1|1,L,R);    else    {        query(left,mid,rt<<1,L,R);        query(mid+1,right,rt<<1|1,L,R);    }}void update(ll left,ll right,ll rt,ll l,ll r,ll up){    if(l<=left&&r>=right)    {        segtree[rt]+=up;        add[rt]+=up;        return;    }    PushDown(rt,right-left+1);    int mid=(left+right)>>1;    if(l<=mid)update(left,mid,rt<<1,l,r,up);    if(r>mid)update(mid+1,right,rt<<1|1,l,r,up);    segtree[rt]=max(segtree[rt<<1],segtree[rt<<1|1]);}int main(){    int t;    scanf("%d",&t);    ll sans,frisk,chara;    int ca=1;    while(t--)    {        scanf("%d%d",&n,&m);        init();        for(int i=0;i<n-1;i++)        {            scanf("%I64d%I64d",&sans,&frisk);            sans++;            frisk++;            adde(sans,frisk);            adde(frisk,sans);        }        for(int i=1;i<=n;i++)        {            scanf("%I64d",&node[i].v);        }        node[1].l=1;        dis[1]=node[1].v;        dfs(1,-1);        plant(1,n,1);        printf("Case #%d:\n",ca++);        for(int i=0;i<m;i++)        {            scanf("%I64d",&chara);            if(chara==1)            {                scanf("%I64d",&frisk);                frisk++;                ans=-1e12;                query(1,n,1,node[frisk].l,node[frisk].r);                printf("%I64d\n",ans);            }            else            {                scanf("%I64d%I64d",&frisk,&sans);                frisk++;                update(1,n,1,node[frisk].l,node[frisk].r,sans-node[frisk].v);                node[frisk].v=sans;            }        }    }    return 0;}
原创粉丝点击