[HDU3966]Aragorn's Story 做题笔记

·· / ·– ·· ·-·· ·-·· / ·–· · ·-· ··· ·· ··· - / ··- -· - ·· ·-·· / ·· / ·– ·· -·

题目来源：http://acm.hdu.edu.cn/showproblem.php?pid=3966

1、打懒标记的时候一定要注意懒标记是否可以累加，或者是否应该异或

2、多组测试数据。。。

3、线段树针对多组数据的初始化不要忘了

#include <cstdio>
#include <algorithm>
#include <cstring>
#define lch now<<1
#define rch now<<1|1
using namespace std;
const int N=60000;
const int inf=0x3fffffff;
int ver[N<<1],e[N<<1],head
,nxt[N<<1];
int n,m,tot=1,cnt=0,L,R;
int dep
,fa
,size
,id
;
int son
,top
,w
,C
;
struct Tree {
int sum,s,del;
}I[N<<2];
void add (int u,int v) {
ver[++tot]=v;nxt[tot]=head[u];head[u]=tot;
}
void pushup (int now) {
I[now].s=I[lch].s+I[rch].s;
I[now].sum=I[lch].sum+I[rch].sum;
}
void pushdown (int now) {
if (I[now].del==0) return ;
I[lch].sum+=I[now].del*I[lch].s;
I[rch].sum+=I[now].del*I[rch].s;
I[lch].del+=I[now].del;//懒标记是可以累加的！
I[rch].del+=I[now].del;//
I[now].del=0;
}
void build (int now,int l,int r) {
if (l==r) {
I[now].sum=C[id[l]];I[now].del=0;
I[now].s=1;
return ;
}
int mid=(l+r)>>1;
build(lch,l,mid); build(rch,mid+1,r);
pushup(now);
}
void change (int now,int l,int r,int del) {
if (l>R||r<L||l>r) return ;
if (L<=l&&r<=R) {
I[now].sum+=del*I[now].s;
I[now].del+=del;//懒标记是可以累加的！
return ;
}
int mid=(l+r)>>1;
pushdown(now);
if (L<=mid) change(lch,l,mid,del);
if (R>mid) change(rch,mid+1,r,del);
pushup(now);
}
int query (int now,int l,int r,int pos) {
if (l==r&&l==pos) return I[now].sum;
int mid=(l+r)>>1;
pushdown(now);
if (pos<=mid) return query(lch,l,mid,pos);//不要忘了return！
if (pos>mid) return query(rch,mid+1,r,pos);
}
void dfs_1 (int u,int from) {
size[u]=1;son[u]=0;fa[u]=from;
for (int i=head[u];i;i=nxt[i]) {
int v=ver[i];
if (v==fa[u]) continue;
dep[v]=dep[u]+1; fa[v]=u;
dfs_1(v,u);size[u]+=size[v];
if (!son[u]||size[son[u]]<size[v]) son[u]=v;
}
}
void dfs_2 (int u,int st) {
w[u]=++cnt;top[u]=st;id[cnt]=u;
if (son[u]) dfs_2(son[u],st);
for (int i=head[u];i;i=nxt[i]) {
int v=ver[i];
if (v==son[u]||v==fa[u]) continue;
dfs_2(v,v);
}
}
void Get (int x,int y,int val) {
while (top[x]!=top[y]) {
if (dep[top[x]]<dep[top[y]])swap(x,y);
L=w[top[x]]; R=w[x];
change(1,1,cnt,val);
x=fa[top[x]];
}
if (dep[x]>dep[y]) swap(x,y);
L=w[x]; R=w[y];
change(1,1,cnt,val);
}
int main () {
int u,v,cap,tmp;
char str[20];
while (scanf("%d%d%d",&n,&tmp,&m)!=EOF) {
tot=1,cnt=0;
memset(head,0,sizeof(head));
memset(size,0,sizeof(size));
memset(son,0,sizeof(son));
memset(fa,0,sizeof(fa));
memset(dep,0,sizeof(dep));
for (int i=0;i<(N<<2);i++) I[i].del=0;//这里初始化一开始忘了
for (int i=1;i<=n;i++) scanf("%d",&C[i]);
for (int i=1;i<n;i++) {
scanf("%d%d",&u,&v);
add(u,v); add(v,u);
}
dfs_1(1,0); dfs_2(1,1);
build(1,1,cnt);
for (int i=1;i<=m;i++) {
scanf("%s",str);
if (str[0]=='I') {
scanf("%d%d%d",&u,&v,&cap);
Get(u,v,cap);
}
else if (str[0]=='D') {
scanf("%d%d%d",&u,&v,&cap);
Get(u,v,-cap);
}
else if (str[0]=='Q') {
scanf("%d",&u);
printf("%d\n",query(1,1,cnt,w[u]));
}
}
}
return 0;
}