一、题目
二、解法
考虑对于
d
e
p
[
l
c
a
(
u
,
v
)
]
dep[lca(u,v)]
dep[lca(u,v)]的另类求法,我们先把
(
1
,
u
)
(1,u)
(1,u)的树上路径权值加
1
1
1,然后对
(
1
,
v
)
(1,v)
(1,v)的点权求和。
这样做有什么好处呢?既然是求和,可以把一大堆东西一起算。我们考虑把一个询问拆成两个,就变了前缀和的询问,我们逐个加入
(
1
,
i
)
(1,i)
(1,i),每一次就顺便处理要求
i
i
i的前缀和的询问,可以用树链剖分维护。
时间复杂度
O
(
n
log
n
)
O(n\log n)
O(nlogn)。
#include <cstdio>
#include <vector>
using namespace std;
const int MAXN = 50005;
const int MOD = 201314;
int read()
{
int x=0,flag=1;char c;
while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
return x*flag;
}
int n,m,cnt,tot,f[MAXN],siz[MAXN],son[MAXN];
int num[MAXN],dep[MAXN],top[MAXN],fa[MAXN],ans[MAXN];
struct edge
{
int v,next;
}e[MAXN*2];
struct node
{
int sum,lazy;
}tr[MAXN*4];
struct data
{
int x,flag,id;
};
vector<data> G[MAXN];
void dfs1(int u)
{
siz[u]++;
dep[u]=dep[fa[u]]+1;
for(int i=f[u];i;i=e[i].next)
{
int v=e[i].v;
if(v==fa[u]) continue;
fa[v]=u;
dfs1(v);
siz[u]+=siz[v];
if(siz[v]>siz[son[u]]) son[u]=v;
}
}
void dfs2(int u,int tp)
{
num[u]=++cnt;top[u]=tp;
if(son[u]) dfs2(son[u],tp);
for(int i=f[u];i;i=e[i].next)
if(e[i].v^fa[u] && e[i].v^son[u])
dfs2(e[i].v,e[i].v);
}
void down(int i,int l,int r)
{
int mid=(l+r)>>1;
tr[i<<1].sum+=(mid-l+1)*tr[i].lazy;
tr[i<<1|1].sum+=(r-mid)*tr[i].lazy;
tr[i<<1].lazy+=tr[i].lazy;
tr[i<<1|1].lazy+=tr[i].lazy;
tr[i].lazy=0;
}
void updata(int i,int l,int r,int L,int R)
{
if(L>r || l>R) return ;
if(L<=l && r<=R)
{
tr[i].sum+=(r-l+1);
tr[i].lazy++;
return ;
}
if(tr[i].lazy) down(i,l,r);
int mid=(l+r)>>1;
updata(i<<1,l,mid,L,R);
updata(i<<1|1,mid+1,r,L,R);
tr[i].sum=tr[i<<1].sum+tr[i<<1|1].sum;
}
int query(int i,int l,int r,int L,int R)
{
if(L>r || l>R) return 0;
if(L<=l && r<=R) return tr[i].sum;
if(tr[i].lazy) down(i,l,r);
int mid=(l+r)>>1;
return (query(i<<1,l,mid,L,R)+query(i<<1|1,mid+1,r,L,R))%MOD;
}
void modify(int u,int v)
{
while(top[u]^top[v])
{
if(dep[top[u]]<dep[top[v]]) swap(u,v);
updata(1,1,n,num[top[u]],num[u]);
u=fa[top[u]];
}
if(dep[u]<dep[v]) swap(u,v);
updata(1,1,n,num[v],num[u]);
}
int ask(int u,int v)
{
int ans=0;
while(top[u]^top[v])
{
if(dep[top[u]]<dep[top[v]]) swap(u,v);
ans+=query(1,1,n,num[top[u]],num[u]),ans%=MOD;
u=fa[top[u]];
}
if(dep[u]<dep[v]) swap(u,v);
ans+=query(1,1,n,num[v],num[u]),ans%=MOD;
return ans;
}
int main()
{
n=read();m=read();
for(int i=2;i<=n;i++)
{
int j=read()+1;
e[++tot]=edge{i,f[j]},f[j]=tot;
e[++tot]=edge{j,f[i]},f[i]=tot;
}
dfs1(1);
dfs2(1,1);
for(int i=1;i<=m;i++)
{
int l=read()+1,r=read()+1,z=read()+1;
G[l-1].push_back(data{z,-1,i});
G[r].push_back(data{z,1,i});
}
for(int i=1;i<=n;i++)
{
modify(1,i);
for(int j=0;j<G[i].size();j++)
{
int x=G[i][j].x,f=G[i][j].flag,id=G[i][j].id;
ans[id]+=f*ask(1,x);
}
}
for(int i=1;i<=m;i++)
printf("%d\n",(ans[i]%MOD+MOD)%MOD);
}