题意
给出一棵数,点有点权。每次询问所有点权在[L,R]之间的点到点u的距离和。强制在线。
n<=150000,Q<=200000
分析
如果没有点权限制的话,我们可以通过树剖线段树,来维护每个点轻儿子子树的节点数量,节点深度和以及节点数量*该点深度的和。
现在有了权值的限制,那么只要把线段树可持久化一下,每次在对应的线段树里面查询即可。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long LL;
const int N=150005;
int n,m,A,a[N],b[N],rt[N],cnt,last[N],dep[N],size[N],top[N],pos[N],tim,fa[N],sz,bot[N],sum_size,q;
LL sum_dep,sum_lca;
struct edge{int to,next,w;}e[N*2];
struct tree{int l,r,s;LL d,dl;}t[N*100];
vector<int> vec[N];
int read()
{
int x=0,f=1;char ch=getchar();
while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
void addedge(int u,int v,int w)
{
e[++cnt].to=v;e[cnt].w=w;e[cnt].next=last[u];last[u]=cnt;
e[++cnt].to=u;e[cnt].w=w;e[cnt].next=last[v];last[v]=cnt;
}
void dfs1(int x)
{
size[x]=1;
for (int i=last[x];i;i=e[i].next)
{
if (e[i].to==fa[x]) continue;
fa[e[i].to]=x;
dep[e[i].to]=dep[x]+e[i].w;
dfs1(e[i].to);
size[x]+=size[e[i].to];
}
}
void dfs2(int x,int chain)
{
pos[x]=++tim;top[x]=chain;bot[x]=x;int k=0;
for (int i=last[x];i;i=e[i].next)
if (e[i].to!=fa[x]&&size[e[i].to]>size[k]) k=e[i].to;
if (!k) return;
dfs2(k,chain);bot[x]=bot[k];
for (int i=last[x];i;i=e[i].next)
if (e[i].to!=fa[x]&&e[i].to!=k) dfs2(e[i].to,e[i].to);
}
void ins(int &d,int l,int r,int x,int y,int z)
{
int p=d;d=++sz;t[d]=t[p];t[d].s++;t[d].d+=y;t[d].dl+=z;
if (l==r) return;
int mid=(l+r)/2;
if (x<=mid) ins(t[d].l,l,mid,x,y,z);
else if (x>mid) ins(t[d].r,mid+1,r,x,y,z);
}
void ins(int id,int x)
{
int w=dep[x];
ins(rt[id],1,n,pos[x],w,dep[x]);
x=fa[top[x]];
while (x)
{
ins(rt[id],1,n,pos[x],w,dep[x]);
x=fa[top[x]];
}
}
void query(int d,int p,int l,int r,int x,int y,int c)
{
if (l==x&&r==y)
{
sum_dep+=c*(t[d].d-t[p].d);
sum_size+=c*(t[d].s-t[p].s);
sum_lca+=c*(t[d].dl-t[p].dl);
return;
}
int mid=(l+r)/2;
if (x<=mid) query(t[d].l,t[p].l,l,mid,x,min(y,mid),c);
if (y>mid) query(t[d].r,t[p].r,mid+1,r,max(x,mid+1),y,c);
}
LL query(int x,int l,int r)
{
int tmp=x;
sum_dep=sum_lca=0;sum_size=0;int y=0;
while (x)
{
LL tmp1=sum_lca;int tmp2=sum_size;sum_lca=sum_size=0;
query(rt[r],rt[l-1],1,n,pos[x],pos[bot[x]],1);
if (y) query(rt[r],rt[l-1],1,n,pos[y],pos[bot[y]],-1);
sum_lca=tmp1+(LL)sum_size*dep[x];sum_size+=tmp2;
if (x!=top[x]) query(rt[r],rt[l-1],1,n,pos[top[x]],pos[x]-1,1);
y=top[x];x=fa[top[x]];
}
return sum_dep+(LL)sum_size*dep[tmp]-sum_lca*2;
}
int main()
{
n=read();q=read();A=read();
for (int i=1;i<=n;i++) a[i]=read(),b[i]=a[i];
sort(b+1,b+n+1);m=unique(b+1,b+n+1)-b-1;
for (int i=1;i<=n;i++) a[i]=lower_bound(b+1,b+m+1,a[i])-b,vec[a[i]].push_back(i);
for (int i=1;i<n;i++)
{
int x=read(),y=read(),z=read();
addedge(x,y,z);
}
dfs1(1);dfs2(1,1);
for (int i=1;i<=m;i++)
{
rt[i]=rt[i-1];
for (int j=0;j<vec[i].size();j++)
ins(i,vec[i][j]);
}
LL ans=0;
while (q--)
{
int u=read(),l=((LL)read()+ans)%A,r=((LL)read()+ans)%A;
if (l>r) swap(l,r);
l=lower_bound(b+1,b+m+1,l)-b;
r=upper_bound(b+1,b+m+1,r)-b-1;
if (l>r) {printf("%d\n",ans=0);continue;}
printf("%lld\n",ans=query(u,l,r));
}
return 0;
}