LOJ #2116 Luogu P3241「HNOI2015」开店

好久没写数据结构了

来补一发

果然写的时候思路极其混乱....

LOJ #2116 Luogu P3241

---

题意

$ Q$次询问，求树上点的颜色在$ [L,R]$中的所有点到询问点的距离 强制在线

询问次数，树上点数约$ 2·10^5$

---

$ Solution$

首先有

$ dist(x,y)=deep(x)+deep(y)-2·deep(lca(x,y))$

显然这个等式的前两项很容易用前缀和什么的维护

只考虑第三项的话相当于是有边权并且强制在线的「LNOI2014」LCA

用同样的套路将$ deep(lca(x,y))$转化成对于所有在区间中的点，将其到根的路径区间加

然后查询询问点到根的距离

将所有点按颜色排序 树剖+主席树维护

每次将当前颜色最小的点插入主席树 最多影响$ log^2$个线段

注意需要标记永久化

写的时候思路很不清楚感觉写了假的主席树

---

$my \ code$

#include<ctime>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#define M 150010
#define rt register int
#define ll long long
using namespace std;
inline ll read(){
    ll x=;char zf=;char ch=getchar();
    while(ch!='-'&&!isdigit(ch))ch=getchar();
    if(ch=='-')zf=-,ch=getchar();
    while(isdigit(ch))x=x*+ch-'',ch=getchar();return x*zf;
}
void write(ll y){if(y<)putchar('-'),y=-y;if(y>)write(y/);putchar(y%+);}
void writeln(const ll y){write(y);putchar('\n');}
int k,m,n,x,y,z,cnt,ans;
int dfn[M],top[M],size[M],fa[M],to[M];ll deep[M];
vector<pair<int,int>>e[];
void dfs(int x,int pre){
    size[x]=;fa[x]=pre;
    for(auto i:e[x])if(i.first!=pre){
        deep[i.first]=deep[x]+i.second;
        dfs(i.first,x);size[x]+=size[i.first];
    }
}
void dfs2(int x,int chain){
    int heavy=;dfn[x]=++cnt;top[x]=chain;to[cnt]=x;
    for(auto i:e[x])if(size[i.first]>size[heavy]&&i.first!=fa[x])heavy=i.first;
    if(!heavy)return;
    dfs2(heavy,chain);
    for(auto i:e[x])if(i.first!=heavy&&i.first!=fa[x])dfs2(i.first,i.first);
}
struct segment{
    int ls,rs,tag;ll sum;
}a[];
int Root[];
#define up a[x].sum=(ll)a[a[x].ls].sum+a[a[x].rs].sum+(ll)a[x].tag*(deep[to[R]]-deep[fa[to[L]]])
void change(int x,int L,int R,int LL,int RR){
    const int mid=L+R>>;
    if(L>=LL&&R<=RR){
        a[x].tag++;
        up;return;
    }
    if(mid>=LL){
        if(a[x].ls)a[cnt+]=a[a[x].ls];
        change(a[x].ls=++cnt,L,mid,LL,RR);
    }
    if(mid+<=RR){
        if(a[x].rs)a[cnt+]=a[a[x].rs];
        change(a[x].rs=++cnt,mid+,R,LL,RR);
    }
    up;
}
ll query(int x,int LL,int RR,int L,int R,int cs){
    if(!x)x=++cnt;if(LL>R||RR<L)return ;
    if(LL>=L&&RR<=R)return a[x].sum+(ll)cs*(deep[to[RR]]-deep[fa[to[LL]]]);
    const int mid=LL+RR>>;
    return query(a[x].ls,LL,mid,L,R,cs+a[x].tag)+query(a[x].rs,mid+,RR,L,R,cs+a[x].tag);
}
void upd(int id,int x){
    while(x){
        change(Root[id],,n,dfn[top[x]],dfn[x]);
        x=fa[top[x]];
    }
}
ll query(int id,int x){
    if(!id)return ;
    ll ans=;
    while(x){
        ans+=query(Root[id],,n,dfn[top[x]],dfn[x],);
        x=fa[top[x]];
    }
    return ans;
}
struct peri{
    int x,id;
    bool operator <(const peri s){
        return x<s.x;
    }
}q[];
ll s[];
int main(){
    n=read();m=read();int A=read();
    for(rt i=;i<=n;i++)q[i].x=read(),q[i].id=i;
    sort(q+,q+n+);fa[]=;
    for(rt i=;i<=n;i++){
        x=read();y=read();z=read();
        e[x].push_back({y,z});e[y].push_back({x,z});
    }
    dfs(,);dfs2(,);cnt=;
    for(rt i=;i<=n;i++)s[i]=s[i-]+deep[q[i].id];//前i小的deep和
    for(rt i=;i<=n;i++){
        Root[i]=++cnt;
        a[Root[i]]=a[Root[i-]];
        upd(i,q[i].id);
    }
    ll las=;
    while(m--){
        z=read();int aa=read(),bb=read();
        int L=min((aa+las)%A,(bb+las)%A);
        int R=max((aa+las)%A,(bb+las)%A);
        L=lower_bound(q+,q+n+,(peri){L,})-q;
        R=lower_bound(q+,q+n+,(peri){R+,})-q-;
        las=(ll)(R-L+)*deep[z]+s[R]-s[L-]-(query(R,z)-query(L-,z))*;
        writeln(las);
    }
    return ;
}