bzoj 4012: [HNOI2015]开店 树链剖分+可持久化线段树

题意

给出一棵数，点有点权。每次询问所有点权在[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;
}