文章标题 HDU 5692 : Snacks（dfn序+线段树）

Snacks

题目链接

分析：首先，先求出每个节点的第一次出现时的dfn序，用in数组维护，第二次出现的dfn序，用out数组来维护，然后，区间【in[x]，out[x]】就是节点的儿子及其自己，所以可以用线段树来维护这段区间的最大值，及题目所求。

代码：

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <queue>
#include <set>
#include <map>
#include <algorithm>
#include <math.h>
#include <vector>
using namespace std;
typedef long long ll;

const int mod=1e9+7;
const int maxn=2e5+10;

const ll inf=1e17;
//first表示节点第一次出现的位置
//in表示第一次出现的dfn序，out表示第二次出现时的dfn序
int in[maxn],out[maxn],idx,first[maxn];
ll dis[maxn];

ll val[maxn];
int n,m;
//建图
struct Edge{
int to,nex;
}edge[maxn];
int tot,head[maxn];
void init(){
tot=0;
idx=0;
memset (head,-1,sizeof (head));
}
void addedge(int u,int v){
edge[tot]=Edge{v,head[u]};head[u]=tot++;
}
//求dfn序，in，out,first
void dfs(int u,int fa){
in[u]=++idx;
first[idx]=u;
for (int i=head[u];i!=-1;i=edge[i].nex){
int v=edge[i].to;
if (v==fa)continue;
dis[v]=dis[u]+val[v];
dfs(v,u);
}
out[u]=idx;
}

ll sum[maxn<<2],setv[maxn<<2];

void push_up(int rt){
sum[rt]=max(sum[rt*2],sum[rt*2+1]);
}

void push_down(int rt){
if (setv[rt]){
setv[rt*2]+=setv[rt];
setv[rt*2+1]+=setv[rt];
sum[rt*2]+=setv[rt];
sum[rt*2+1]+=setv[rt];
setv[rt]=0;
}
}

void build (int rt,int l,int r){//建树
setv[rt]=0;
if (l==r){
sum[rt]=dis[first[l]];//第一次出现是时的节点的距离
return;
}
int mid=(l+r)/2;
build(rt*2,l,mid);
build(rt*2+1,mid+1,r);
push_up(rt);
}

void update(int rt,int l,int r,int ul,int ur,int val){//更新
if(ul<=l&&ur>=r){
sum[rt]+=val;
setv[rt]+=val;
return;
}
push_down(rt);
int mid=(r+l)/2;
if (ul<=mid)update(rt*2,l,mid,ul,ur,val);
if (ur>mid)update(rt*2+1,mid+1,r,ul,ur,val);
push_up(rt);
}

ll query(int rt,int l,int r,int ql,int qr){//查询
if (ql<=l&&qr>=r){
return sum[rt];
}
push_down(rt);
int mid=(l+r)/2;
ll ans=-inf;
if (ql<=mid)ans=max(ans,query(rt*2,l,mid,ql,qr));
if (qr>mid)ans=max(ans,query(rt*2+1,mid+1,r,ql,qr));
push_up(rt);
return ans;
}

int main()
{
int T;
scanf ("%d",&T);
int cas=1;
while (T--){
init();
scanf ("%d%d",&n,&m);
int u,v;
for (int i=0;i<n-1;i++){
scanf ("%d%d",&u,&v);
addedge(u,v);
addedge(v,u);
}
for (int i=0;i<n;i++){
scanf ("%lld",&val[i]);
}
dis[0]=val[0];
dfs(0,-1);
build(1,1,n);
//      for (int i=0;i<n;i++){
//          printf ("dis %d = %d\n",i,dis[i]);
//      }
int op;
printf ("Case #%d:\n",cas++);
while (m--){
scanf ("%d",&op);
if (op==1){
scanf ("%d",&u);
ll ans=query(1,1,n,in[u],out[u]);
printf ("%lld\n",ans);
}else {
scanf("%d%d",&u,&v);
update(1,1,n,in[u],out[u],v-val[u]);
val[u]=v;
}
}
}
return 0;
}