题目:http://acm.hdu.edu.cn/showproblem.php?pid=5692
题意:
Problem 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零食机的路线中,价值总和的最大值。
思路:2016百度之星初赛题目,当时以为是树链剖分 + 线段树,但是不会做-_-||。。。原来是dfs序 + 线段树。首先把树形转换为线性,我们用dfs序给每个点编号,in[v] 到 out[v]之间的编号包括了v的所有子孙,这道题就是求0到v的子孙的最大距离,于是变成了查询区间最大值,更新时区间更新。
总结:这题long long数据是个坑。。。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <queue>
using namespace std;
typedef long long ll;
const int N = 100010;
const ll INF = 1e18;
struct edge
{
int to, next;
}g[N*2];
struct node
{
int l, r;
ll val, mark;
}s[N*4];
int head[N], in[N], out[N];
ll tmp[N], val[N];
int n, m, cnt, num, _case = 0;
void add_edge(int v, int u)
{
g[cnt].to = u;
g[cnt].next = head[v];
head[v] = cnt++;
}
void dfs(int v, int fa, ll w)
{
in[v] = ++num, val[num] = w;
for(int i = head[v]; i != -1; i = g[i].next)
{
int u = g[i].to;
if(u != fa)
dfs(u, v, w + tmp[u]);
}
out[v] = num;
}
void push_up(int k)
{
s[k].val = max(s[k<<1].val, s[k<<1|1].val);
}
void push_down(int k)
{
if(s[k].mark)
{
s[k<<1].val += s[k].mark;
s[k<<1|1].val += s[k].mark;
s[k<<1].mark += s[k].mark;
s[k<<1|1].mark += s[k].mark;
s[k].mark = 0;
}
}
void build(int l, int r, int k)
{
s[k].l = l, s[k].r = r, s[k].mark = 0;
if(l == r)
{
s[k].val = val[l];
return;
}
int mid = (l + r) >> 1;
build(l, mid, k << 1);
build(mid + 1, r, k << 1|1);
push_up(k);
}
void update(int l, int r, ll c, int k)
{
if(l <= s[k].l && s[k].r <= r)
{
s[k].val += c;
s[k].mark += c;
return;
}
push_down(k);
int mid = (s[k].l + s[k].r) >> 1;
if(l <= mid) update(l, r, c, k << 1);
if(r > mid) update(l, r, c, k << 1|1);
push_up(k);
}
ll query(int l, int r, int k)
{
if(l <= s[k].l && s[k].r <= r)
return s[k].val;
push_down(k);
int mid = (s[k].l + s[k].r) >> 1;
ll ans = -INF;
if(l <= mid) ans = max(ans, query(l, r, k << 1));
if(r > mid) ans = max(ans, query(l, r, k << 1|1));
return ans;
}
int main()
{
int t, a, b;
ll c;
scanf("%d", &t);
while(t--)
{
scanf("%d%d", &n, &m);
cnt = num = 0;
memset(head, -1, sizeof head);
for(int i = 0; i < n - 1; i++)
{
scanf("%d%d", &a, &b);
add_edge(a, b);
add_edge(b, a);
}
for(int i = 0; i < n; i++)
scanf("%I64d", &tmp[i]);
dfs(0, -1, tmp[0]);
build(1, num, 1);
printf("Case #%d:\n", ++_case);
for(int i = 0; i < m; i++)
{
scanf("%d%d", &a, &b);
if(a == 1) printf("%I64d\n", query(in[b], out[b], 1));
else
{
scanf("%I64d", &c);
update(in[b], out[b], c - tmp[b], 1);
tmp[b] = c;
}
}
}
return 0;
}