HDU 5692 Snacks bfs版本dfs序 线段树

Snacks

题目连接：

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

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

Hint

题意

题解：

第一个想法去熟练剖分了，导致智障了一下……

每个点维护从根到这个点的权值和

其实就是dfs序维护子树就好了，修改这个点的值，只会修改子树的值

然后输出这个子树的最大值就好了……

注意，这个傻逼题会爆栈，所以得用bfs去写dfs序

代码

#include<stdio.h>
#include<algorithm>
#include<stack>
#include<queue>
#include<iostream>
#include<cstring>
#include<math.h>
using namespace std;
const int maxn = 1e5+7;
inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
typedef long long SgTreeDataType;
struct treenode
{
int L , R  ;
SgTreeDataType sum , lazy;
void update(SgTreeDataType v)
{
sum += v;
lazy += v;
}
};

treenode tree[maxn*4];

inline void push_down(int o)
{
SgTreeDataType lazyval = tree[o].lazy;
tree[2*o].update(lazyval) ; tree[2*o+1].update(lazyval);
tree[o].lazy = 0;
}

inline void push_up(int o)
{
tree[o].sum = max(tree[2*o].sum , tree[2*o+1].sum);
}

inline void build_tree(int L , int R , int o)
{
tree[o].L = L , tree[o].R = R,tree[o].sum = tree[o].lazy = 0;
if (R > L)
{
int mid = (L+R) >> 1;
build_tree(L,mid,o*2);
build_tree(mid+1,R,o*2+1);
}
}

inline void update(int QL,int QR,SgTreeDataType v,int o)
{
int L = tree[o].L , R = tree[o].R;
if (QL <= L && R <= QR) tree[o].update(v);
else
{
push_down(o);
int mid = (L+R)>>1;
if (QL <= mid) update(QL,QR,v,o*2);
if (QR >  mid) update(QL,QR,v,o*2+1);
push_up(o);
}
}

inline SgTreeDataType query(int QL,int QR,int o)
{
int L = tree[o].L , R = tree[o].R;
if (QL <= L && R <= QR) return tree[o].sum;
else
{
push_down(o);
int mid = (L+R)>>1;
SgTreeDataType res = -(1LL<<62);
if (QL <= mid) res = max(res,query(QL,QR,2*o));
if (QR > mid) res = max(res,query(QL,QR,2*o+1));
push_up(o);
return res;
}
}
vector<int> E[maxn];
int in[maxn],out[maxn],fa[maxn],n,m;
long long Ex[maxn],val[maxn],cnt;
stack<int> S;
vector<int> V;
void getdfs()
{
S.push(1);
while(!S.empty())
{
int now = S.top();
S.pop();
V.push_back(now);
in[now]=out[now]=++cnt;
for(int i=0;i<E[now].size();i++)
{
int v = E[now][i];
if(v==fa[now])continue;
fa[v]=now;
Ex[v]=Ex[now]+val[v];
S.push(v);
}
}
}
void init()
{
for(int i=0;i<maxn;i++)E[i].clear();
cnt=0;
V.clear();
while(!S.empty())S.pop();
memset(in,0,sizeof(in));
memset(out,0,sizeof(out));
memset(fa,0,sizeof(fa));
}
void solve(int cas)
{
init();
scanf("%d%d",&n,&m);
for(int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
x++,y++;
E[x].push_back(y);
E[y].push_back(x);
}
for(int i=1;i<=n;i++)scanf("%lld",&val[i]);
Ex[1]=val[1];
build_tree(1,n,1);
getdfs();
reverse(V.begin(),V.end());
for(int i=0;i<V.size();i++)
{
int p = V[i];
update(in[p],in[p],Ex[p],1);
for(int j=0;j<E[p].size();j++)
{
int v = E[p][j];
if(v==fa[p])continue;
out[p]=max(out[p],out[v]);
}
}
printf("Case #%d:\n",cas);
while(m--)
{
int ppp = read(),x = read();
x++;
if(ppp==0){
long long y;
scanf("%lld",&y);
update(in[x],out[x],y-val[x],1);
val[x]=y;
}
else
printf("%lld\n",query(in[x],out[x],1));
}
}
int main()
{
int t;
scanf("%d",&t);
for(int i=1;i<=t;i++)solve(i);
return 0;
}