未来城市规划------线段树+树链剖分

原题
其实就是利用树链剖分建图,然后在线段树上实现。树链剖分中的每一个树枝都可以在线段数中有对应的。然后,至于算出‘城市u所管辖下的所有城市对’的花费之和,其实就是每一个子城市(v)的贡献=路程wsiz[v](siz[u]-siz[v])(这个很容易证),可以在线段树中用2个数字维护。一个维护wsiz[v],另一个维护wsiz[v]*siz[v],然后就是区间求和,区间改变。

#include<bits/stdc++.h>
#define lson k<<1,l,mid
#define rson k<<1|1,mid+1,r
#define ll long long
using namespace std;
const int MX=5e4+9;
const int mod=2019;
struct node{
    int u,v,next;
}edge[MX];

int n,q,u,v,futh,cnt=0,tot=0;
int head[MX],fa[MX],siz[MX],tp[MX],son[MX],de[MX],po[MX],st[MX],ed[MX];
int w,a[MX],t1[MX<<2],t2[MX<<2],laze[MX<<2],s1[MX<<2],s2[MX<<2];
char s[20];
void add(int u,int v){
    edge[cnt].v=u;
    edge[cnt].next=head[v];
    head[v]=cnt++;
}

void dfs1(int u,int fath,int deth){
    fa[u]=fath,de[u]=deth,son[u]=-1,siz[u]=1;
    for( int i=head[u] ; ~i ; i=edge[i].next ){
        if( edge[i].v==fath )
            continue;
        dfs1(edge[i].v,u,deth+1);
        siz[u]+=siz[edge[i].v];
        if( son[u]==-1 || siz[son[u]]<siz[edge[i].v] )
            son[u]=edge[i].v;
    }
}

void dfs2(int u,int top){
    st[u]=++tot;
    po[tot]=u;
    tp[u]=top;
    if( son[u]>0 )
        dfs2(son[u],top);
    for( int i=head[u] ; ~i ; i=edge[i].next )
        if( edge[i].v!=son[u] )
            dfs2(edge[i].v,edge[i].v);
    ed[u]=tot;
    return ;
}

void pushup(int k){
    t1[k]=(t1[k<<1]+t1[k<<1|1])%mod;
    t2[k]=(t2[k<<1]+t2[k<<1|1])%mod;
}

void pushdown(int k){
    if( laze[k] ){
        t1[k<<1]=(t1[k<<1]+(laze[k]*s1[k<<1])%mod)%mod;
        t1[k<<1|1]=(t1[k<<1|1]+(laze[k]*s1[k<<1|1])%mod)%mod;
        t2[k<<1]=(t2[k<<1]+(laze[k]*s2[k<<1])%mod)%mod;
        t2[k<<1|1]=(t2[k<<1|1]+(laze[k]*s2[k<<1|1])%mod)%mod;
        laze[k<<1]=(laze[k<<1]+laze[k])%mod;
        laze[k<<1|1]=(laze[k<<1|1]+laze[k])%mod;
        laze[k]=0;
    }
}

void build(int k,int l,int r){
    laze[k]=0;
    if( l==r ){
        s1[k]=siz[po[l]]%mod;
        s2[k]=s1[k]*s1[k]%mod;
        t1[k]=s1[k]*a[po[l]]%mod;
        t2[k]=s2[k]*a[po[l]]%mod;
        return ;
    }
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(k);
    s1[k]=(s1[k<<1]+s1[k<<1|1])%mod;
    s2[k]=(s2[k<<1]+s2[k<<1|1])%mod;
}

void update(int k,int l,int r,int L,int R,int w){
    if( L<=l && r<=R ){
        t1[k]=(t1[k]+w*s1[k]%mod)%mod;
        t2[k]=(t2[k]+w*s2[k]%mod)%mod;
        laze[k]=(laze[k]+w)%mod;
        return ;
    }
    pushdown(k);
    int mid=(l+r)>>1;
    if( L<=mid )
        update(lson,L,R,w);
    if( mid<R )
        update(rson,L,R,w);
    pushup(k);
}

int que1(int k,int l,int r,int L,int R){
    if( L<=l && r<=R )
        return t1[k];
    pushdown(k);
    int mid=(l+r)>>1;
    int ans=0;
    if( L<=mid )
        ans+=que1(lson,L,R);
    if( mid<R )
        ans+=que1(rson,L,R);
    return ans%mod;
}

int que2(int k,int l,int r,int L,int R){
    if( L<=l && r<=R )
        return t2[k];
    pushdown(k);
    int mid=(l+r)>>1;
    int ans=0;
    if( L<=mid )
        ans+=que2(lson,L,R);
    if( mid<R )
        ans+=que2(rson,L,R);
    return ans%mod;
}

void vs(int u,int v,ll w){
    int f1=tp[u],f2=tp[v];
    while( f1!=f2 ){
        if( de[f1]>de[f2] ){
            swap(f1,f2);
            swap(u,v);
        }
        update(1,1,n,st[f2],st[v],w);   // 走一次,更新一次
        v=fa[f2];  
        f2=tp[v];
    }
    if( u==v )
        return ;
    if( st[u]>st[v] )
        swap(u,v);
    update(1,1,n,st[u]+1,st[v],w);
    return ;
}

int main()
{
   // freopen("input.txt","r",stdin);
    scanf("%d %d",&n,&q);
    memset(head,-1,sizeof(head));
    for( int i=1 ; i<=n-1 ; i++ ){
        scanf("%d %d",&futh,&w);
        a[i+1]=w;
        add(i+1,futh);
    }
    dfs1(1,0,1);
    dfs2(1,1);
    build(1,1,n);
    while( q-- ){
        scanf("%s",s);
        if( s[0]=='I' ){
            scanf("%d %d %d",&u,&v,&w);
            vs(u,v,w%mod);
        }
        else{
            scanf("%d",&u);
            if( siz[u]==1 )
                printf("0\n");
            else{
                int q1=que1(1,1,n,st[son[u]]+1,ed[u]),q2=que2(1,1,n,st[son[u]]+1,ed[u]);  //记住,例如2到3,改变的其实只有3,没有2(谁的深度低,谁被改变)
                int ans=(q1*siz[u]-q2+mod)%mod;
                printf("%d\n",ans%mod);
            }
        }
    }
    fclose(stdin);fclose(stdout);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值