百度科技园内有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;
}