HDU5692(线段树+dfs序)

时间:2023-12-26 23:52:31

Snacks

Time Limit:5000MS     Memory Limit:65536KB     64bit IO Format:%I64d & %I64u
Submit Status

Description

百度科技园内有n个零食机,零食机之间通过n−1条路相互连通。每个零食机都有一个值v,表示为小度熊提供零食的价值。

由于零食被频繁的消耗和补充,零食机的价值v会时常发生变化。小度熊只能从编号为0的零食机出发,并且每个零食机至多经过一次。另外,小度熊会对某个零食机的零食有所偏爱,要求路线上必须有那个零食机。

为小度熊规划一个路线,使得路线上的价值总和最大。 

Input

输入数据第一行是一个整数T(T≤10),表示有T组测试数据。

对于每组数据,包含两个整数n,m(1≤n,m≤100000),表示有n个零食机,m次操作。

接下来n−1行,每行两个整数x和y(0≤x,y<n),表示编号为x的零食机与编号为y的零食机相连。

接下来一行由n个数组成,表示从编号为0到编号为n−1的零食机的初始价值v(|v|<100000)。

接下来m行,有两种操作:0 x y,表示编号为x的零食机的价值变为y;1 x,表示询问从编号为0的零食机出发,必须经过编号为x零食机的路线中,价值总和的最大值。

本题可能栈溢出,辛苦同学们提交语言选择c++,并在代码的第一行加上:

`#pragma comment(linker, "/STACK:1024000000,1024000000") `

Output

对于每组数据,首先输出一行”Case #?:”,在问号处应填入当前数据的组数,组数从1开始计算。

对于每次询问,输出从编号为0的零食机出发,必须经过编号为x零食机的路线中,价值总和的最大值。 

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序
 //2016.8.11
#pragma comment(linker, "/STACK:1024000000, 1024000000")
#include<iostream>
#include<cstdio>
#include<vector>
#include<algorithm>
#define N 100005
#define lson (id<<1)
#define rson ((id<<1)|1)
#define mid ((l+r)>>1) using namespace std; struct node
{
long long max, lazy;
}tree[N*]; vector<int> v[N];
long long w[N], val[N];//val[i]表示从根节点走到i节点的价值和
int in[N], out[N], flag;//in记录查询区间的左端,out记录查询区间的右端 //线段树模板
//************************************************************************************
void push_up(int id)
{
tree[id].max = max(tree[lson].max, tree[rson].max);
return;
} void build_tree(int id, int l, int r)
{
if(l == r)
{
tree[id].max = val[l];
tree[id].lazy = ;
return ;
}
build_tree(lson, l, mid);
build_tree(rson, mid+, r);
push_up(id);
tree[id].lazy = ;
return ;
} void push_down(int id, int l, int r)
{
if(tree[id].lazy)
{
tree[lson].lazy += tree[id].lazy;
tree[rson].lazy += tree[id].lazy;
tree[lson].max += tree[id].lazy;
tree[rson].max += tree[id].lazy;
tree[id].lazy = ;
}
return ;
} void ins(int id, int l, int r, int ql, int qr, int tt)
{
if(ql<=l&&r<=qr)
{
tree[id].max += tt;
tree[id].lazy += tt;
return;
}
if(tree[id].lazy) push_down(id, l, r);
if(ql<=mid)
ins(lson, l, mid, ql, qr, tt);
if(mid+<=qr)
ins(rson, mid+, r, ql, qr, tt);
push_up(id);
return;
} long long query(int id, int l, int r, int ql, int qr)
{
if(ql<=l && r<=qr)
{
return tree[id].max;
}
if(tree[id].lazy)
push_down(id, l, r);
long long maxnum = -;
if(ql<=mid) maxnum = max(maxnum, query(lson, l, mid, ql, qr));
if(mid+<=qr) maxnum = max(maxnum, query(rson, mid+, r, ql, qr));
return maxnum;
}
//************************************************************************************ void dfs(int a, int fa, long long wight)
{
in[a] = ++flag;
for(int i = ; i < v[a].size(); i++)
{
int b = v[a][i];
if(b==fa)continue;
dfs(b, a, w[b]+wight);
}
out[a] = flag;
val[in[a]] = wight;
} int main()
{
int T, n, m, a, b, cmd;
cin>>T;
for(int kase = ; kase <= T; kase++)
{
printf("Case #%d:\n", kase);
cin>>n>>m;
for(int i = ; i < N; i++)
v[i].clear();
for(int i = ; i < n-; i++)
{
scanf("%d%d", &a, &b);
a++; b++;
v[a].push_back(b);
v[b].push_back(a);
}
for(int i = ; i <= n; i++)
scanf("%lld", &w[i]);
flag = ;
dfs(, , w[]);
for(int i = ; i <= n; i++)
cout<<in[i]<<" "<<out[i]<<endl;
build_tree(, , n);
while(m--)
{
scanf("%d", &cmd);
if(cmd == )
{
scanf("%d%d", &a, &b);a++;
ins(, , n, in[a], out[a], b-w[a]);
w[a] = b;
}else
{
scanf("%d", &a); a++;
long long ans = query(, , n, in[a], out[a]);
cout<<ans<<endl;
}
}
} return ;
}