4034: [HAOI2015]树上操作
Time Limit: 10 Sec Memory Limit: 256 MBSubmit: 3553 Solved: 1118
[ Submit][ Status][ Discuss]
Description
有一棵点数为 N 的树,以点 1 为根,且树点有边权。然后有 M 个
操作,分为三种:
操作 1 :把某个节点 x 的点权增加 a 。
操作 2 :把某个节点 x 为根的子树中所有点的点权都增加 a 。
操作 3 :询问某个节点 x 到根的路径中所有点的点权和。
Input
第一行包含两个整数 N, M 。表示点数和操作数。接下来一行 N 个整数,表示树中节点的初始权值。接下来 N-1
行每行三个正整数 fr, to , 表示该树中存在一条边 (fr, to) 。再接下来 M 行,每行分别表示一次操作。其中
第一个数表示该操作的种类( 1-3 ) ,之后接这个操作的参数( x 或者 x a ) 。
Output
对于每个询问操作,输出该询问的答案。答案之间用换行隔开。
Sample Input
5 5
1 2 3 4 5
1 2
1 4
2 3
2 5
3 3
1 2 1
3 5
2 1 2
3 3
1 2 3 4 5
1 2
1 4
2 3
2 5
3 3
1 2 1
3 5
2 1 2
3 3
Sample Output
6
9
13
9
13
HINT
对于 100% 的数据, N,M<=100000 ,且所有输入数据的绝对值都不会超过 10^6 。
Source
题解:树链剖分
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define N 200003
#define LL long long
using namespace std;
int tot,n,m,next[N],point[N],v[N],deep[N],fa[N],size[N],son[N];
LL tr[N*4],delta[N*4];
int val[N],l[N],r[N],q[N],pos[N],cnt,belong[N];
void add(int x,int y)
{
tot++; next[tot]=point[x]; point[x]=tot; v[tot]=y;
tot++; next[tot]=point[y]; point[y]=tot; v[tot]=x;
}
void dfs(int x,int f)
{
deep[x]=deep[f]+1;
size[x]=1;
for (int i=point[x];i;i=next[i])
if (v[i]!=f) {
fa[v[i]]=x;
dfs(v[i],x);
size[x]+=size[v[i]];
if (size[son[x]]<size[v[i]]) son[x]=v[i];
}
}
void dfs1(int x,int chain)
{
belong[x]=chain; pos[x]=++cnt; l[x]=r[x]=cnt; q[cnt]=x;
if (!son[x]) return;
dfs1(son[x],chain);
for (int i=point[x];i;i=next[i])
if (v[i]!=fa[x]&&v[i]!=son[x])
dfs1(v[i],v[i]);
r[x]=cnt;
}
void update(int now)
{
tr[now]=tr[now<<1]+tr[now<<1|1];
}
void build(int now,int l,int r)
{
if (l==r) {
tr[now]=(LL)val[q[l]];
return;
}
int mid=(l+r)/2;
build(now<<1,l,mid);
build(now<<1|1,mid+1,r);
update(now);
}
void pushdown(int now,int l,int r)
{
int mid=(l+r)/2;
if (delta[now]) {
tr[now<<1]+=(LL)(mid-l+1)*delta[now];
tr[now<<1|1]+=(LL)(r-mid)*delta[now];
delta[now<<1]+=delta[now];
delta[now<<1|1]+=delta[now];
delta[now]=0;
}
}
void pointchange(int now,int l,int r,int x,int v)
{
if (l==r) {
tr[now]+=(LL)v;
return;
}
int mid=(l+r)/2;
pushdown(now,l,r);
if (x<=mid) pointchange(now<<1,l,mid,x,v);
else pointchange(now<<1|1,mid+1,r,x,v);
update(now);
}
void qjchange(int now,int l,int r,int ll,int rr,int v)
{
if (ll<=l&&r<=rr) {
tr[now]+=(LL)v*(LL)(r-l+1);
delta[now]+=(LL)v;
return;
}
int mid=(l+r)/2;
pushdown(now,l,r);
if (ll<=mid) qjchange(now<<1,l,mid,ll,rr,v);
if (rr>mid) qjchange(now<<1|1,mid+1,r,ll,rr,v);
update(now);
}
LL qjsum(int now,int l,int r,int ll,int rr)
{
if (ll<=l&&r<=rr) return tr[now];
int mid=(l+r)/2; LL ans=0;
pushdown(now,l,r);
if (ll<=mid) ans+=qjsum(now<<1,l,mid,ll,rr);
if (rr>mid) ans+=qjsum(now<<1|1,mid+1,r,ll,rr);
return ans;
}
LL solve(int x,int y)
{
LL ans=0;
while (belong[x]!=belong[y]) {
if (deep[belong[x]]<deep[belong[y]]) swap(x,y);
ans+=qjsum(1,1,n,pos[belong[x]],pos[x]);
x=fa[belong[x]];
}
if(deep[x]>deep[y]) swap(x,y);
ans+=qjsum(1,1,n,pos[x],pos[y]);
return ans;
}
int main()
{
freopen("a.in","r",stdin);
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++) scanf("%d",&val[i]);
for (int i=1;i<n;i++) {
int x,y; scanf("%d%d",&x,&y);
add(x,y);
}
dfs(1,0); dfs1(1,1);
build(1,1,n);
for (int i=1;i<=m;i++) {
int opt,x,y; scanf("%d",&opt);
if (opt==1) {
scanf("%d%d",&x,&y);
pointchange(1,1,n,pos[x],y);
}
if (opt==2) {
scanf("%d%d",&x,&y);
qjchange(1,1,n,l[x],r[x],y);
}
if (opt==3) {
scanf("%d",&x);
printf("%I64d\n",solve(1,x));
}
}
}