Night_sea_64 @ 2024-02-08 22:57:10
rt,下面的
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
int n,m,root,mod,val[100010];
int sz[100010],dep[100010],fa[100010],son[100010];
int id[100010],cur,pos[100010],top[100010];
vector<int>v[100010];
void pre(int x,int last)
{
sz[x]=1,dep[x]=dep[last]+1,fa[x]=last;
for(auto y:v[x])
if(y!=last)
{
pre(y,x);
sz[x]+=sz[y];
if(sz[y]>sz[son[x]])son[x]=y;
}
}
void dfs(int x,int last,int tp)
{
id[++cur]=x,pos[x]=cur,top[x]=tp;
if(son[x])dfs(son[x],x,tp);
for(auto y:v[x])
if(y!=last&&y!=son[x])dfs(y,x,y);
}
struct segtree{int l,r,sum,lazy;}a[600010];
void build(int x,int l,int r)
{
a[x].l=l,a[x].r=r;
if(l==r)
{
a[x].sum=val[id[l]];
return;
}
int mid=(l+r)/2;
build(x*2,l,mid);
build(x*2+1,mid+1,r);
a[x].sum=(a[x*2].sum+a[x*2+1].sum)%mod;
}
void add(int x,int l,int r,int k)
{
if(a[x].l==l&&a[x].r==r)
{
a[x].lazy=(a[x].lazy+k)%mod;
return;
}
a[x].sum=(a[x].sum+1ll*(r-l+1)*k%mod)%mod;
int mid=(a[x].l+a[x].r)/2;
if(l<=mid)add(x*2,l,min(r,mid),k);
if(r>mid)add(x*2+1,max(mid+1,l),r,k);
}
void pushdown(int x)
{
a[x].sum=(a[x].sum+1ll*(a[x].r-a[x].l+1)*a[x].lazy%mod)%mod;
a[x*2].lazy=(a[x*2].lazy+a[x].lazy)%mod;
a[x*2+1].lazy=(a[x*2+1].lazy+a[x].lazy)%mod;
a[x].lazy=0;
}
int query(int x,int l,int r)
{
pushdown(x);
if(a[x].l==l&&a[x].r==r)return a[x].sum;
int mid=(a[x].l+a[x].r)/2,ret=0;
if(l<=mid)ret=(ret+query(x*2,l,min(r,mid)))%mod;
if(r>mid)ret=(ret+query(x*2+1,max(mid+1,l),r))%mod;
return ret;
}
void addline(int x,int y,int k)
{
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]])swap(x,y);
add(1,pos[top[x]],pos[x],k);
x=fa[top[x]];
}
if(dep[x]>dep[y])swap(x,y);
add(1,pos[x],pos[y],k);
}
void addtree(int x,int k){
add(1,pos[x],pos[x]+sz[x]-1,k);
}
int queryline(int x,int y)
{
int ret=0;
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]])swap(x,y);
ret=(ret+query(1,pos[top[x]],pos[x]))%mod,x=fa[top[x]];
}
if(dep[x]>dep[y])swap(x,y);
return (ret+query(1,pos[x],pos[y]))%mod;
}
int querytree(int x){
return query(1,pos[x],pos[x]+sz[x]-1);
}
int main()
{
cin>>n>>m>>root>>mod;
for(int i=1;i<=n;i++)cin>>val[i];
for(int i=1;i<n;i++)
{
int x,y;
cin>>x>>y;
v[x].push_back(y);
v[y].push_back(x);
}
pre(root,0);
dfs(root,0,root);
build(1,1,n);
while(m--)
{
int op;
cin>>op;
if(op==1)
{
int x,y,z;
cin>>x>>y>>z;
addline(x,y,z);
}
if(op==2)
{
int x,y;
cin>>x>>y;
cout<<queryline(x,y)<<endl;
}
if(op==3)
{
int x,z;
cin>>x>>z;
addtree(x,z);
}
if(op==4)
{
int x;
cin>>x;
cout<<querytree(x)<<endl;
}
}
return 0;
}
by Terrible @ 2024-02-08 23:17:16
@Night_sea_64
这个 query
上来就 pushdown
吗?
by Composite_Function @ 2024-02-09 11:29:29
你 query
中没有判断边界条件就下传标记了,导致叶子节点也会下传,所以四倍空间不够。
by Night_sea_64 @ 2024-02-09 18:53:53
@Terrible @Composite_Function
谢谢大佬!此帖结
by Night_sea_64 @ 2024-02-09 18:55:19
惊现 fgx 大佬(