我这奇怪的线段树四倍空间咋还不够啊!??

P3384 【模板】重链剖分/树链剖分

Night_sea_64 @ 2024-02-08 22:57:10

rt,下面的 a 数组需要开到 600000 才能 AC。

#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 大佬(


|