题目:http://acm.hdu.edu.cn/showproblem.php?pid=3966
题意:
给一棵树,并给定各个点权的值,然后有3种操作:
I C1 C2 K: 把C1与C2的路径上的所有点权值加上K
D C1 C2 K:把C1与C2的路径上的所有点权值减去K
Q C:查询节点编号为C的权值
分析:
树链剖分入门题,基于点的重编号。
操作是区间增减+单点查询
用树状数组或者线段树维护操作
代码:
树状数组
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <cstdio>
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
#include <string>
#include <map>
#include <cmath>
#include <queue>
#include <set>
using namespace std;
const int N = 50010;
struct EDGE {
int v,nex;
} edge[N<<1];
int head[N],tot;
int dep[N],p[N],fp[N],fa[N],top[N],son[N],siz[N];
int cnt;
int num[N];
void addedge(int a,int b) {
edge[tot].v=b;
edge[tot].nex=head[a];
head[a]=tot++;
}
void dfs(int u) {
siz[u]=1,son[u]=0;
for(int i=head[u]; ~i; i=edge[i].nex) {
int v=edge[i].v;
if(v!=fa[u]) {
fa[v]=u;
dep[v]=dep[u]+1;
dfs(v);
if(siz[v]>siz[son[u]])son[u]=v;
siz[u]+=siz[v];
}
}
}
void build(int u,int tp) {
p[u]=++cnt;
fp[p[u]]=u;
top[u]=tp;
if(son[u])build(son[u],tp);
for(int i=head[u]; ~i; i=edge[i].nex) {
int v=edge[i].v;
if(v!=son[u]&&v!=fa[u])build(v,v);
}
}
//树状数组
int arr[N];
inline int lowbit(int x){return x&(-x);}
inline int sum(int x) {
int res=0;
while(x)res+=arr[x],x-=lowbit(x);
return res;
}
inline void add(int x,int n) {
while(x<N)arr[x]+=n,x+=lowbit(x);
}
inline int update(int x,int y,int n) {
add(x,n);
add(y+1,-n);
}
void change(int a,int b,int x) {
int f1=top[a],f2=top[b],tmp=0;
while(f1!=f2) {
if(dep[f1]<dep[f2])swap(f1,f2),swap(a,b);
update(p[f1],p[a],x);
a=fa[f1],f1=top[a];
}
// if(a==b)return;
if(dep[a]>dep[b])swap(a,b);
update(p[a],p[b],x);
}
int n,m,q,a,b,c;
char op[10];
int main() {
int T;
//freopen("f.txt","r",stdin);
while(~scanf("%d%d%d",&n,&m,&q)) {
memset(head,-1,sizeof(head));
memset(arr,0,sizeof(arr));
cnt=tot=0;
for(int i=1; i<=n; i++)scanf("%d",&num[i]);
for(int i=0; i<m; i++) {
scanf("%d%d",&a,&b);
addedge(a,b);
addedge(b,a);
}
dfs(1);
build(1,1);
for(int i=1;i<=n;i++)update(p[i],p[i],num[i]);
while(q--) {
scanf("%s",op);
if(op[0]=='Q') {
scanf("%d",&a);
printf("%d\n",sum(p[a]));
} else {
scanf("%d%d%d",&a,&b,&c);
if(op[0]=='D')c=-c;
change(a,b,c);
}
}
}
return 0;
}
线段树:
这题用线段树不如树状数组简单,但是学习下套路还是挺不错的
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <cstdio>
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
#include <string>
#include <map>
#include <cmath>
#include <queue>
#include <set>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int N = 50010;
struct EDGE {
int v,nex;
} edge[N<<1];
int head[N],tot;
int dep[N],p[N],fp[N],fa[N],top[N],son[N],siz[N];
int cnt,sum[N<<2],col[N<<2];
int num[N];
void addedge(int a,int b) {
edge[tot].v=b;
edge[tot].nex=head[a];
head[a]=tot++;
}
void dfs(int u) {
siz[u]=1,son[u]=0;
for(int i=head[u]; ~i; i=edge[i].nex) {
int v=edge[i].v;
if(v!=fa[u]) {
fa[v]=u;
dep[v]=dep[u]+1;
dfs(v);
if(siz[v]>siz[son[u]])son[u]=v;
siz[u]+=siz[v];
}
}
}
void build(int u,int tp) {
p[u]=++cnt;
fp[p[u]]=u;
top[u]=tp;
if(son[u])build(son[u],tp);
for(int i=head[u]; ~i; i=edge[i].nex) {
int v=edge[i].v;
if(v!=son[u]&&v!=fa[u])build(v,v);
}
}
//线段树
//void pushUp(int rt) {
// sum[rt]=sum[rt<<1]+sum[rt<<1|1];
//}
void pushDown(int rt,int m) {
if(col[rt]) {
col[rt<<1]+=col[rt];
col[rt<<1|1]+=col[rt];
sum[rt<<1]+=(m-(m>>1))*col[rt];
sum[rt<<1|1]+=(m>>1)*col[rt];
col[rt]=0;
}
}
void build(int l,int r,int rt) {
col[rt]=0;
if(l==r) {
sum[rt]=num[fp[l]];
return;
}
int m=l+r>>1;
build(lson);
build(rson);
// pushUp(rt);
}
int query(int p,int l,int r,int rt) {
if(l==r) {
return sum[rt];
}
pushDown(rt,r-l+1);
int m=l+r>>1;
int ret=0;
if(p<=m)ret=query(p,lson);
else ret=query(p,rson);
// pushUp(rt);
return ret;
}
void update(int a,int b,int x,int l,int r,int rt) {
if(a<=l&&r<=b) {
col[rt]+=x;
sum[rt]+=x*(r-l+1);
return;
}
pushDown(rt,r-l+1);
int m=l+r>>1;
if(a<=m)update(a,b,x,lson);
if(b>m)update(a,b,x,rson);
//pushUp(rt);
}
void change(int a,int b,int x) {
int f1=top[a],f2=top[b],tmp=0;
while(f1!=f2) {
if(dep[f1]<dep[f2])swap(f1,f2),swap(a,b);
update(p[f1],p[a],x,1,cnt,1);
a=fa[f1],f1=top[a];
}
// if(a==b)return;
if(dep[a]>dep[b])swap(a,b);
update(p[a],p[b],x,1,cnt,1);
}
int n,m,q,a,b,c;
char op[10];
int main() {
int T;
// freopen("f.txt","r",stdin);
while(~scanf("%d%d%d",&n,&m,&q)) {
memset(head,-1,sizeof(head));
cnt=tot=0;
for(int i=1; i<=n; i++)scanf("%d",&num[i]);
for(int i=0; i<m; i++) {
scanf("%d%d",&a,&b);
addedge(a,b);
addedge(b,a);
}
dfs(1);
build(1,1);
build(1,n,1);
while(q--) {
scanf("%s",op);
if(op[0]=='Q') {
scanf("%d",&a);
printf("%d\n",query(p[a],1,n,1));
} else {
scanf("%d%d%d",&a,&b,&c);
if(op[0]=='D')c=-c;
change(a,b,c);
}
}
}
return 0;
}