[LNOI2014]LCA

https://www.luogu.org/problem/P4211

可以发现,问题能够转化为从询问点到根都加1,然后询问l到r的点到根的路径上权值之和。
然后这个问题又和“l到r的点到根路径上所有点权值分别+1,求询问点到根路径上的权值之和”等价。
进一步,发现询问是可减的,即等价于ans[r]-ans[l-1]。
把询问按编号排序
用树链剖分+线段树实现链修改+链查询。
边修改边回答询问即可。
注意答案要对201314取模。

#include<iostream>
#include<cctype>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<cstdlib>
#include<algorithm>
#define N 220000
#define eps 1e-7
#define inf 1e9+7
#define ll long long
using namespace std;
inline int read()
{
	char ch=0;
	int x=0,flag=1;
	while(!isdigit(ch)){ch=getchar();if(ch=='-')flag=-1;}
	while(isdigit(ch)){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
	return x*flag;
}
struct edge
{
	int to,nxt;
}e[N];
int num,head[N];
inline void add(int x,int y)
{
	e[++num]=(edge){y,head[x]};
	head[x]=num;
}
struct node
{
	int x,t,id,flag;
}p[N];
bool cmp(node a,node b)
{
	return a.t<b.t;
}
struct Segment_Tree
{
	#define lson o<<1
	#define rson o<<1|1
	#define mid ((l+r)>>1)
	int addv[N*4],sumv[N*4];
	inline void pushup(int o)
	{
		sumv[o]=sumv[lson]+sumv[rson];
	}
	inline void color(int o,int l,int r,int num)
	{
		addv[o]+=num;
		sumv[o]+=(r-l+1)*num;
	}
	inline void pushdown(int o,int l,int r)
	{
		color(lson,l,mid,addv[o]);
		color(rson,mid+1,r,addv[o]);
		addv[o]=0;
	}
	int query(int o,int l,int r,int ql,int qr)
	{
		if(ql<=l&&r<=qr)return sumv[o];
		pushdown(o,l,r);
		int ans=0;
		if(ql<=mid)ans+=query(lson,l,mid,ql,qr);
		if(qr>mid)ans+=query(rson,mid+1,r,ql,qr);
		return ans; 
	}
	void optadd(int o,int l,int r,int ql,int qr)
	{
		if(ql<=l&&r<=qr)
		{
			color(o,l,r,1);
			return;
		}
		pushdown(o,l,r);
		if(ql<=mid)optadd(lson,l,mid,ql,qr);
		if(qr>mid)optadd(rson,mid+1,r,ql,qr);
		pushup(o);
	}
}T;
int n,m,q,times,id[N],fa[N],sz[N],son[N],dep[N],top[N],ans[N];
void dfs1(int x,int t)
{
	sz[x]=1;dep[x]=t;
	for(int i=head[x];i;i=e[i].nxt)
	{
		int to=e[i].to;
		dfs1(to,t+1);
		fa[to]=x;sz[x]+=sz[to];
		if(sz[son[x]]<sz[to])son[x]=to;
	}
}
void dfs2(int x,int tp)
{
	top[x]=tp;id[x]=++times;
	if(son[x])dfs2(son[x],tp);
	for(int i=head[x];i;i=e[i].nxt)
	{
		int to=e[i].to;
		if(to==son[x])continue;
		dfs2(to,to);
	}
}
int qrange(int x,int y)
{
	int ans=0;
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		ans+=T.query(1,1,n,id[top[x]],id[x]);
		x=fa[top[x]];
	}
	if(dep[x]>dep[y])swap(x,y);
	ans+=T.query(1,1,n,id[x],id[y]);
	return ans;
}
void optrange(int x,int y)
{
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		T.optadd(1,1,n,id[top[x]],id[x]);
		x=fa[top[x]];
	}
	if(dep[x]>dep[y])swap(x,y);
	T.optadd(1,1,n,id[x],id[y]);
	return;
}
int main()
{
	int i,j,l,r,x;
	n=read();q=read();
	for(i=2;i<=n;i++)add(read()+1,i);
	dfs1(1,1);dfs2(1,1);
	for(i=1;i<=q;i++)
	{
		l=read()+1;r=read()+1;x=read()+1;
		p[++m]=(node){x,l-1,i,-1};
		p[++m]=(node){x,r+0,i,+1};
	}
	sort(p+1,p+m+1,cmp);
	for(i=1,j=0;i<=m;i++)
	{
		while(j<n&&j<p[i].t)optrange(1,++j);
		ans[p[i].id]+=p[i].flag*qrange(1,p[i].x);
	}
	for(i=1;i<=q;i++)printf("%d\n",ans[i]%201314);
	return 0;
}
posted @ 2018-12-05 01:12  Creed-qwq  阅读(124)  评论(0编辑  收藏  举报