比较恶心，特别是线段树
题目链接：P3384 【模板】重链剖分
详解见：大佬博客
我的代码：
\(Code\):
#include
#include
using namespace std;
const int MAXN=100005;
int mod,n,m,root;
int f[MAXN],dep[MAXN],son[MAXN],tot[MAXN];
int t[MAXN],b[MAXN],id[MAXN],c=0,top[MAXN];
struct edge
{
int to,nxt;
}e[MAXN<<1];
int head[MAXN],cnt=0;
void swap(int *a,int *b){int t=*a;*a=*b,*b=t;}
void add(int u,int v)
{
e[++cnt].to=v;
e[cnt].nxt=head[u];
head[u]=cnt;
}
int dfs1(int cur,int fa,int deep)
{
dep[cur]=deep;
f[cur]=fa;
int maxn=-1;
tot[cur]=1;
for(int i=head[cur];i;i=e[i].nxt)
{
int j=e[i].to;
if(!(j^fa)) continue;
tot[cur]+=dfs1(j,cur,deep+1);
if(tot[j]>maxn) maxn=tot[j],son[cur]=j;
}
return tot[cur];
}
void dfs2(int cur,int topf)
{
id[cur]=++c;
t[c]=b[cur];
top[cur]=topf;
if(!son[cur]) return;
dfs2(son[cur],topf);
for(int i=head[cur];i;i=e[i].nxt)
{
int j=e[i].to;
if(!id[j]) dfs2(j,j);
}
}
struct node
{
int l,r,sum,lazy;
node()
{
l=r=sum=lazy=0;
}
}a[MAXN<<2];
void update(int k){a[k].sum=(a[k<<1].sum+a[k<<1|1].sum)%mod;}
void build(int k,int l,int r)
{
int mid=(l+r)>>1;
a[k].l=l,a[k].r=r;
if(l==r)
{
a[k].sum=t[l];
return;
}
build(k<<1,l,mid),build(k<<1|1,mid+1,r);
update(k);
}
void lazydown(int k)
{
if(a[k].l==a[k].r)
{
a[k].lazy=0;
return;
}
a[k<<1].sum=(a[k<<1].sum+(a[k<<1].r-a[k<<1].l+1)*a[k].lazy%mod)%mod;
a[k<<1|1].sum=(a[k<<1|1].sum+(a[k<<1|1].r-a[k<<1|1].l+1)*a[k].lazy%mod)%mod;
a[k<<1].lazy=(a[k<<1].lazy+a[k].lazy)%mod;
a[k<<1|1].lazy=(a[k<<1|1].lazy+a[k].lazy)%mod;
a[k].lazy=0;
}
void change(int k,int l,int r,int x)
{
int mid=(a[k].l+a[k].r)>>1;
if(a[k].l==l&&a[k].r==r)
{
a[k].sum=(a[k].sum+(a[k].r-a[k].l+1)*x%mod)%mod;
a[k].lazy=(a[k].lazy+x)%mod;
return;
}
if(a[k].lazy) lazydown(k);
if(r<=mid) change(k<<1,l,r,x);
else if(l>mid) change(k<<1|1,l,r,x);
else change(k<<1,l,mid,x),change(k<<1|1,mid+1,r,x);
update(k);
}
int query(int k,int l,int r)
{
int mid;
if(a[k].lazy) lazydown(k);
if(a[k].l==l&&a[k].r==r) return a[k].sum%mod;
mid=(a[k].l+a[k].r)>>1;
if(r<=mid) return query(k<<1,l,r)%mod;
if(l>=mid+1) return query(k<<1|1,l,r)%mod;
return (query(k<<1,l,mid)+query(k<<1|1,mid+1,r))%mod;
}
void tree_add(int l,int r,int x)
{
while(top[l]!=top[r])
{
if(dep[top[l]]dep[r]) swap(l,r);
change(1,id[l],id[r],x);
return;
}
int tree_query(int l,int r)
{
int ans=0;
while(top[l]!=top[r])
{
if(dep[top[l]]dep[r]) swap(l,r);
ans=(ans+query(1,id[l],id[r]))%mod;
return ans%mod;
}
void son_add(int k,int x){change(1,id[k],id[k]+tot[k]-1,x);}
int son_query(int k){return query(1,id[k],id[k]+tot[k]-1);}
int u,v,flag,x,y,z;
int main()
{
scanf("%d%d%d%d",&n,&m,&root,&mod);
for(int i=1;i<=n;i++) scanf("%d",&b[i]);
for(int i=1;i 还是写炸了线段树，因为左右端点反了只有\(30pts\)（数据水？）
还看了\(bi\)站视频，爽。
主要思想是根据\(dfs\)序使链和子树映射在线段树上，使得计算的相关点重新编号后在一个区间中，可证明时间复杂度为\(O(mlog^2n)\)，似乎懂了。
就这样