【LNOI2014】LCA【差分】【树剖】

4 篇文章 0 订阅
4 篇文章 0 订阅

叶子最可爱啦qwq!

每次询问一个区间和一个点,求这个区间所有点和给定点的LCA的深度和。

深度和是吧。

LCA的深度和有一个很优良的性质。你把一个点到根的权值都+1,查另一个点到根的权值就是这个深度。

那问题转化成了链加链求和。

不过有q个询问哦。怎么办呢?

很明显这个询问应该被去掉。

我们不一个询问一个询问查询。

我们直接从1号点开始加入贡献。对于一个询问,加到l-1时统计一下,加到r时统计一下,然后两者减减就行。

利用了差分的思想。

当然是树剖+线段树。

WARNING:不要像我一样,跳top时交换了top没交换自己。

#include<bits/stdc++.h>
using namespace std;
#define in read()
int in{
	int cnt=0,f=1;char ch=0;
	while(!isdigit(ch)){
		ch=getchar();if(ch=='-')f=-1;
	}
	while(isdigit(ch)){
		cnt=cnt*10+ch-48;
		ch=getchar();
	}return cnt*f;
}
const int mod=201314;
struct node{
	int l,r,key,tag;
}t[200003];
int n,m;
int first[100003],nxt[200003],to[200003],tot;
void add(int a,int b){
	nxt[++tot]=first[a];first[a]=tot;to[tot]=b;
}
int id[100003],top[100003],son[100003],size[100003],dep[100003],fa[100003];
void pushup(int u){
	t[u].key=t[u*2].key+t[u*2+1].key;t[u].key%=mod;
}
void pushdown(int u){
	if(t[u].tag){
		t[u*2].tag+=t[u].tag;t[u*2+1].tag+=t[u].tag;t[u*2].tag%=mod;t[u*2+1].tag%=mod;
		t[u*2].key+=(t[u*2].r-t[u*2].l+1)*t[u].tag;t[u*2].key%=mod;
		t[u*2+1].key+=(t[u*2+1].r-t[u*2+1].l+1)*t[u].tag;t[u*2+1].key%=mod;
		t[u].tag=0;
	}
}
int queryseg(int u,int ql,int qr){
	if(t[u].r<ql||t[u].l>qr)return 0;
	if(ql<=t[u].l&&t[u].r<=qr)return t[u].key;
	pushdown(u);
	return (queryseg(u*2,ql,qr)+queryseg(u*2+1,ql,qr))%mod;
}
void segmo(int u,int ql,int qr,int key){
	if(t[u].r<ql||t[u].l>qr)return;
	if(ql<=t[u].l&&t[u].r<=qr){
		//cout<<t[u].l<<" "<<t[u].r<<endl;
		t[u].key+=t[u].r-t[u].l+1;t[u].key%=mod;t[u].tag+=key;t[u].tag%=mod;return;
	}pushdown(u);segmo(u*2,ql,qr,key);segmo(u*2+1,ql,qr,key);
	pushup(u);
}
void modify(int a,int b){
	//cout<<"@# "<<a<<" "<<b<<" "<<top[a]<<" "<<top[b]<<endl; 
	int x=top[a],y=top[b];
	while(x!=y){
		if(dep[x]<dep[y])swap(x,y),swap(a,b);
		segmo(1,id[x],id[a],1);//cout<<"# "<<x<<" "<<a<<endl;
		a=fa[x];x=top[a];
	}
	if(id[a]>id[b])swap(a,b);//cout<<"# "<<a<<" "<<b<<endl;
	segmo(1,id[a],id[b],1);
}int cntt;
int query(int a,int b){
	int x=top[a],y=top[b];int gu=0;
	while(x!=y){
		if(dep[x]<dep[y])swap(x,y),swap(a,b);
		gu+=queryseg(1,id[x],id[a]);
		a=fa[x];x=top[a];gu%=mod;
	}
	if(id[a]>id[b])swap(a,b);
	gu+=queryseg(1,id[a],id[b]);return gu%mod;
}
struct bili{
	int flag,pos,ans,belong,z;
}que[100003];int cnt;

int ans1[100003],ans2[100003];
void dfs1(int u,int faa){
	dep[u]=dep[faa]+1;size[u]=1;fa[u]=faa;
	for(int i=first[u];i;i=nxt[i]){
		int v=to[i];if(v==faa)continue;
		dfs1(v,u);size[u]+=size[v];
		if(size[v]>size[son[u]])son[u]=v;
	} 
}
void dfs2(int u,int tp){
	top[u]=tp;id[u]=++cntt;
	if(son[u])dfs2(son[u],tp);
	for(int i=first[u];i;i=nxt[i]){
		int v=to[i];if(v==fa[u]||v==son[u])continue;dfs2(v,v);
	}
}
void build(int u,int l,int r){
	t[u].l=l;t[u].r=r;
	if(l==r){
		return;
	}int mid=(l+r)>>1;
	build(u*2,l,mid);build(u*2+1,mid+1,r);
}
bool cm(bili a,bili b){
	return a.pos<b.pos;
}
int ans[100003];
void debug(int u,int l,int r){
	cout<<l<<" "<<r<<" "<<t[u].key<<endl;
	if(l==r){return;}pushdown(u);
	int mid=(l+r)>>1;
	debug(u*2,l,mid);debug(u*2+1,mid+1,r);
}
int main(){
	n=in;m=in;
	for(int i=2;i<=n;i++){
		int x=in+1;add(x,i);
	}
	build(1,1,n);dfs1(1,0);dfs2(1,1);
	//for(int i=1;i<=n;i++)cout<<id[i]<<" ";cout<<endl;
	//debug(1,1,n);
	for(int i=1;i<=m;i++){
		int l=in;int r=in;int z=in;++r;++z;
		que[++cnt].pos=l;que[cnt].belong=i;que[cnt].z=z;
		que[++cnt].pos=r;que[cnt].belong=i;que[cnt].flag=1;que[cnt].z=z;
	}
	sort(que+1,que+cnt+1,cm);
	for(int i=1,pos=0;i<=cnt;i++){
		while(pos<que[i].pos){
			modify(1,++pos);//cout<<pos<<endl;
			//debug(1,1,n);//cout<<"###########"<<endl;
		}
		que[i].ans=query(1,que[i].z);
		if(que[i].flag)(ans[que[i].belong]+=que[i].ans)%=mod;
		else ((ans[que[i].belong]-=que[i].ans)+=mod)%=mod;
	}
	for(int i=1;i<=m;i++)printf("%lld\n",ans[i]);
	
	return 0;	
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值