社区讨论
树剖求救,虽然我觉得不会有人回复我QwQ
P3384【模板】重链剖分 / 树链剖分参与者 3已保存回复 4
讨论操作
快速查看讨论及其快照的属性,并进行相关操作。
- 当前回复
- 4 条
- 当前快照
- 1 份
- 快照标识符
- @mi7yhmgl
- 此快照首次捕获于
- 2025/11/21 05:41 4 个月前
- 此快照最后确认于
- 2025/11/21 05:41 4 个月前
CPP
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<cmath>
#define int long long
using namespace std;
const int N=5000005;
int e=0,ver[N<<1],nexts[N<<1],head[N];
void add(int x,int y){ver[++e]=y,nexts[e]=head[x],head[x]=e;}
struct Tc{int dep,fa,size,son,id,top;}Tcut[N];
int cnt;
struct Tree{int dat,add,l,r;}t[N<<2];
int val[N],tval[N];
int n,m,root,mol;
void dfs1(int x,int fa,int deep)
{
Tcut[x].size=1;
Tcut[x].dep=deep;
Tcut[x].fa=fa;
int Maxson=-1;
for(int i=head[x];i;i=nexts[i])
{
int v=ver[i];
if(v==fa)continue;
dfs1(v,x,deep+1);
Tcut[x].size+=Tcut[v].size;
if(Maxson<Tcut[v].size){Tcut[x].son=v; Maxson=Tcut[v].size;}
}
}
void dfs2(int x,int top)
{
Tcut[x].id=++cnt;
tval[cnt]=val[x]%mol;
Tcut[x].top=top;
if(!Tcut[x].son)return;
dfs2(Tcut[x].son,top);
for(int i=head[x];i;i=nexts[i])
{
int v=ver[i];
if(v==Tcut[x].fa||v==Tcut[x].son)continue;
dfs2(v,v);
}
}
void pushup(int p){t[p].dat=t[p<<1].dat+t[p<<1|1].dat;t[p].dat%=mol;}
void build(int p,int l,int r)
{
t[p].l=l;t[p].r=r;
if(l==r){t[p].dat=tval[l];t[p].dat%=mol;return;}
int mid=l+r>>1;
build(p<<1,l,mid);build(p<<1|1,mid+1,r);
pushup(p);
}
void pushdown(int p)
{
if(t[p].add)
{
t[p<<1].add+=t[p].add;t[p<<1|1].add+=t[p].add;
t[p<<1].dat+=t[p].add*(t[p<<1].r-t[p<<1].l+1);
t[p<<1|1].dat+=t[p].add*(t[p<<1|1].r-t[p<<1|1].l+1);
t[p<<1].add%=mol,t[p<<1|1].add%=mol,t[p<<1].dat%=mol,t[p<<1|1].dat%=mol;
t[p].add=0;
}
}
int query(int p,int l,int r)
{
if(t[p].l>=l&&r>=t[p].r)return t[p].dat;
pushdown(p);
int mid=t[p].l+t[p].r>>1;
int ans=0;
if(l<=mid)ans+=query(p<<1,l,r),ans%=mol;
if(mid<r)ans+=query(p<<1|1,l,r),ans%=mol;
return ans%mol;
}
void change(int p,int l,int r,int v)
{
if(l<=t[p].l&&r>=t[p].r)
{
t[p].add+=v;
t[p].dat+=v*(t[p].r-t[p].l+1);
t[p].add%=mol;
t[p].dat%=mol;
return;
}
int mid=t[p].l+t[p].r>>1;
pushdown(p);
if(mid>=l)change(p<<1,l,r,v);
if(r>mid)change(p<<1|1,l,r,v);
pushup(p);
}
int query_edge(int x,int y)
{
int ans=0;
while(Tcut[x].top!=Tcut[y].top)
{
if(Tcut[x].dep<Tcut[y].dep)swap(x,y);
ans+=query(1,Tcut[x].id,Tcut[Tcut[x].top].id);
ans%=mol;
x=Tcut[Tcut[x].top].fa;
}
if(Tcut[x].dep>Tcut[y].dep)swap(x,y);
ans+=query(1,Tcut[x].id,Tcut[y].id);
return ans%mol;
}
void change_edge(int x,int y,int z)
{
z%=mol;
while(Tcut[x].top!=Tcut[y].top)
{
if(Tcut[x].top<Tcut[y].top)swap(x,y);
change(1,Tcut[x].id,Tcut[Tcut[x].top].id,z);
x=Tcut[Tcut[x].top].fa;
}
if(Tcut[x].dep>Tcut[y].dep)swap(x,y);
change(1,Tcut[x].id,Tcut[y].id,z);
}
int query_p(int x)
{
return query(1,Tcut[x].id,Tcut[x].id+Tcut[x].size-1)%mol;
}
void change_p(int x,int z)
{
change(1,Tcut[x].id,Tcut[x].id+Tcut[x].size-1,z);
}
signed main()
{
scanf("%lld%lld%lld%lld",&n,&m,&root,&mol);
for(int i=1;i<=n;i++)scanf("%lld",&val[i]);
for(int i=1;i<n;i++)
{
int x,y;
scanf("%lld%lld",&x,&y);
add(x,y);add(y,x);
}
dfs1(root,0,1);dfs2(root,root);build(1,1,n);
for(int i=1;i<=m;i++)
{
int k,x,y,z;
scanf("%lld",&k);
if(k==1){scanf("%lld%lld%lld",&x,&y,&z);change_edge(x,y,z);}
else if(k==2){scanf("%lld%lld",&x,&y);printf("%lld\n",query_edge(x,y)%mol);}
else if(k==3){scanf("%lld%lld",&x,&z);change_p(x,z);}
else scanf("%lld",&x),printf("%lld\n",query_p(x)%mol);
}
return 0;
}
回复
共 4 条回复,欢迎继续交流。
正在加载回复...