牛客网比赛212E蚂蚁开会题解

这个题目其实就是【HNOI2015开店】的主席树做法,但稍有不同十分类似。

题目地址【IN


  • 题意简述:

给你一棵树,有点权和边权,每次有两种操作一种询问:

  • 修改一个点权
  • 修改一条边权
  • 询问所有的点到给定点uu的距离值。
    一个点vvuu的距离值为vv的点权乘以vuv\rightarrow u的边权。

暴力就暴力改和统计吧,没什么好说的,也没什么部分分。


  • 正解
    Orz大巨佬hdxrie的讲解

这个题不能用动态点分治,是因为它对于点权和边权都有大量的修改,并且它每次询问的是所有的点到某个点的值,所以我们不用动态点分治。

我们这样处理边权:将其下放到一个点上,按照根节点下面到深度较深的点上。

然后我们像开店那个题一样,对于答案就是下方式子的值:

ans=i=1nval[i]×dis[i]+i=1nval[i]×dis[u]2×i=1nval[i]×dis[lca(i,u)]ans=\sum\limits_{i=1}^nval[i]\times dis[i]+\sum_{i=1}^nval[i]\times dis[u]-2\times \sum_{i=1}^nval[i]\times dis[lca(i,u)]

由于对于点的要求没有,不用主席树,所以直接如下维护:
对于i=1nval[i]×dis[i]\sum\limits_{i=1}^nval[i]\times dis[i]i=1nval[i]×dis[lca(i,u)]\sum_{i=1}^nval[i]\times dis[lca(i,u)]我们用线段树和树链剖分来维护,对于dis[u]dis[u]我们用树状数组和深搜序来维护,对于i=1nval[i]\sum_{i=1}^nval[i]我们开一个变量allall维护即可。


线段树:

我们维护三个,sz[o],len[o],sum[o]sz[o],len[o],sum[o]sz[o]sz[o]表示当前oo表示的区间的点权修改懒标记,len[o]len[o]表示当前区间的点所表示的边长之和,sum[o]sum[o]表示当前区间的边长乘以点权的和。

  • 对于修改点权,我们可以通过像点分治一样的方式,跳轻重链,跳到根节点修改,每次将变化量加到懒标记上,然后更新贡献即可,复杂度log2nlog^2n
  • 对于边的修改,我们相当于修改它的一个单点,所以单点修改即可,复杂度lognlogn

树状数组:

我们就维护一个区间修改单点查询的树状数组即可,对于一条边的修改,它只会影响到它的子树内的点到根节点的距离,所以用深搜序区间修改即可。复杂度2logn2logn

全局变量allall

这个只是表示点权之和,所以每次修改先减去之前的点权加上新的点权即可。复杂度O(1)O(1)


答案查询

第一个就是线段树的根节点值sum[1]sum[1],第二个就是all×dis[u]all\times dis[u]全局变量乘以单点查询的uu到根节点的值,然后就是lca(i,u)lca(i,u)的查询,也就是从点uu到根节点上,维护的sum[i]sum[i]的和即可,因为lca(i,u)lca(i,u)肯定为uu或者uu的祖先,所以直接查询到根节点的一条链上的sum[i]sum[i]的和就是答案。

对于查询lca(i,u)lca(i,u)的答案那一段为什么是正确的,下面简单说明一下:

因为每个lca(i,u)lca(i,u)到根节点的距离都会被lca(i,u)lca(i,u)为根的子树内的点走一次,所以这样直接加上就是答案,而不会少算或者算重。

t1=sum[1],t2=all×query(u),t3=getans(u)t1=sum[1],t2=all\times query(u),t3=getans(u),答案就为t1+t22×t3t1+t2-2\times t3


代码(略微有点长和丑):
注意略去自环。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int M=2e5+10;
int n,m;
ll side[M],P[M];
struct ss{
	int to,last,len;
	ss(){}
	ss(int a,int b,int c):to(a),last(b),len(c){}
}g[M<<1];
int head[M],cnt;
void add(int a,int b,int c){
	g[++cnt]=ss(b,head[a],c);head[a]=cnt;
	g[++cnt]=ss(a,head[b],c);head[b]=cnt;
}
#define lowbit(a) ((a)&(-(a)))
ll dis[M],all;
int dep[M],f[M],top[M],sze[M],son[M];
int dfn[M],tim,rf[M],sp[M];
ll tot[M];
void add(int a,ll b){
	for(;a<=n;a+=lowbit(a))dis[a]+=b;
}
void add_area(int l,int r,ll b){
	if(l>r)swap(l,r);
	add(l,b);add(r+1,-b);
}
void dfs1(int a,ll lenn){
	sze[a]=1;dfn[a]=++tim;tot[a]=P[a];
	add_area(tim,tim,lenn);
	for(int i=head[a];i;i=g[i].last){
		if(g[i].to==f[a]||g[i].to==a) continue;
		f[g[i].to]=a;
		dep[g[i].to]=dep[a]+1;
		dfs1(g[i].to,lenn+g[i].len);
		tot[a]+=tot[g[i].to];
		sze[a]+=sze[g[i].to];
		if(!son[a]||sze[son[a]]<sze[g[i].to])son[a]=g[i].to;
	}
}
void dfs2(int a,int b){
	top[a]=b;sp[a]=++tim;rf[tim]=a;
	if(!son[a]) return;
	dfs2(son[a],b);
	for(int i=head[a];i;i=g[i].last){
		if(g[i].to==a||g[i].to==f[a]||g[i].to==son[a]) continue;
		dfs2(g[i].to,g[i].to);
	}
}
ll query(int a){
	int ans=0;
	for(;a;a-=lowbit(a)) ans+=dis[a];
	return ans;
}
ll sz[M<<2],len[M<<2],sum[M<<2];
void pushup(int o){
	len[o]=len[o<<1]+len[o<<1|1];
	sum[o]=sum[o<<1]+sum[o<<1|1];
}
void pushdown(int o){
	if(!sz[o]) return;
	sz[o<<1]+=sz[o];
	sz[o<<1|1]+=sz[o];
	sum[o<<1]+=len[o<<1]*sz[o];
	sum[o<<1|1]+=len[o<<1|1]*sz[o];
	sz[o]=0;
}
void build(int o,int l,int r){
	if(l==r){
		sz[o]=tot[rf[l]];len[o]=side[rf[l]];
		sum[o]=sz[o]*len[o];
		return;
	}
	int mid=l+r>>1;
	build(o<<1,l,mid);
	build(o<<1|1,mid+1,r);
	pushup(o);
}
void update_pot(int o,int l,int r,int L,int R,ll p){
	if(L<=l&&r<=R){
		sz[o]+=p;
		sum[o]+=p*len[o];
		return;
	}
	int mid=l+r>>1;
	pushdown(o);
	if(L<=mid) update_pot(o<<1,l,mid,L,R,p);
	if(R>mid) update_pot(o<<1|1,mid+1,r,L,R,p);
	pushup(o);
}
void update_sid(int o,int l,int r,int p,ll v){
	if(l==r){
		len[o]=v;
		sum[o]=v*sz[o];
		return;
	}
	pushdown(o);
	int mid=l+r>>1;
	if(p<=mid) update_sid(o<<1,l,mid,p,v);
	else update_sid(o<<1|1,mid+1,r,p,v);
	pushup(o);
}
ll query(int o,int l,int r,int L,int R){
	if(L<=l&&r<=R) return sum[o];
	int mid=l+r>>1;ll ans=0;
	pushdown(o);
	if(L<=mid) ans=query(o<<1,l,mid,L,R);
	if(R>mid) ans+=query(o<<1|1,mid+1,r,L,R);
	return ans;
}
ll getans(int a){
	ll now=0;
	for(int i=a;i;i=f[top[i]])
		now+=query(1,1,n,sp[top[i]],sp[i]);
	return now;
}
void add_pot(int a,ll b){
	for(int i=a;i;i=f[top[i]]){
		update_pot(1,1,n,sp[top[i]],sp[i],b-P[a]);
	}
	all+=b-P[a];	
	P[a]=b;
}
void add_sid(int a,ll b){
	++a;
	add_area(dfn[a],dfn[a]+sze[a]-1,b-side[a]);
	update_sid(1,1,n,sp[a],b);
	side[a]=b;
}
void answer(int p){
	ll t1=sum[1];
	ll t2=1ll*all*query(dfn[p]);
	ll t3=getans(p);
	printf("%lld\n",t1+t2-(t3<<1));
}
int a,x;
ll b;
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%lld",&P[i]),all+=P[i];
	for(int i=1;i<n;i++){
		scanf("%d%lld",&a,&b);
		add(a,i+1,b);side[i+1]=b;f[i+1]=a;
	}
	dfs1(1,0);
	tim=0;
	dfs2(1,1);
	build(1,1,n);
	while(m--){
		scanf("%d",&a);
		if(a==1){
			scanf("%d%lld",&x,&b);
			add_pot(x,b);			
		}else if(a==2){
			scanf("%d%lld",&x,&b);
			add_sid(x,b);
		}else if(a==3){
			scanf("%d",&x);
			answer(x);
		}
	}
	return 0;
}
posted @ 2018-10-23 14:00  VictoryCzt  阅读(128)  评论(0编辑  收藏  举报