标签：剖分 int top scc 树链 son id 模板 define

#include<bits/stdc++.h>
#define N 100010
#define INF 0x3f3f3f3f
#define eps 1e-10
#define pi 3.141592653589793
#define P 1000000007 
#define LL long long
#define pb push_back
#define fi first
#define se second
#define cl clear
#define si size
#define lb lower_bound
#define ub upper_bound
#define mem(x) memset(x,0,sizeof x)
#define sc(x) scanf("%d",&x)
#define scc(x,y) scanf("%d%d",&x,&y)
#define sccc(x,y,z) scanf("%d%d%d",&x,&y,&z)
using namespace std;
vector<int> a[N];
int n,m,cnt,rt,tg,son[N],top[N],id[N],fa[N],d[N],sz[N],rk[N],w[N],mod;
struct node{ int s,lz; }f[N<<2];
void dfs1(int x,int ffa){
    sz[x]=1;
    for(auto i:a[x]) if (i!=ffa){
        fa[i]=x; d[i]=d[x]+1; dfs1(i,x); sz[x]+=sz[i];
        if (sz[i]>sz[son[x]]) son[x]=i;
    }
}
void dfs2(int x,int t){
    top[x]=t; id[x]=++cnt; rk[cnt]=x;
    if (!son[x]) return;
    dfs2(son[x],t);
    for (auto i:a[x]) if (i!=son[x] && i!=fa[x]) dfs2(i,i);
}

inline void pushdown(int x,int l,int r,int t){
    if (f[x].lz){
        f[x<<1].lz+=f[x].lz; f[x<<1].s+=(t-l+1)*f[x].lz;
        f[x<<1|1].lz+=f[x].lz; f[x<<1|1].s+=(r-t)*f[x].lz;
        f[x].lz=0;
    }
}

void updata(int x,int l,int r,int fl,int fr){
    if (l==fl && r==fr){
        f[x].lz+=tg; f[x].s+=(r-l+1)*tg;
    }else{
        int t=l+r>>1;
        pushdown(x,l,r,t);
        if (fr<=t) updata(x<<1,l,t,fl,fr);else
        if (fl>t) updata(x<<1|1,t+1,r,fl,fr);else{
            updata(x<<1,l,t,fl,t);
            updata(x<<1|1,t+1,r,t+1,fr);
        }
        f[x].s=f[x<<1].s+f[x<<1|1].s;
    }
}

int query(int x,int l,int r,int fl,int fr){
    if (l==fl && r==fr) return f[x].s;
    int t=l+r>>1;
    pushdown(x,l,r,t);
    if (fr<=t)return query(x<<1,l,t,fl,fr);else
    if (fl>t)return query(x<<1|1,t+1,r,fl,fr);else
    return query(x<<1,l,t,fl,t)+query(x<<1|1,t+1,r,t+1,fr);
}

int sum(int x,int y){
    int ans=0;
    while(top[x]!=top[y]){
        if (d[top[x]]<d[top[y]]) swap(x,y);
        ans+=query(1,1,n,id[top[x]],id[x]);
        x=fa[top[x]];
    }
    if (d[x]>d[y]) swap(x,y);
    ans+=query(1,1,n,id[x],id[y]);
    return ans;
}
void change(int x,int y){
    while(top[x]!=top[y]){
        if (d[top[x]]<d[top[y]]) swap(x,y);
        updata(1,1,n,id[top[x]],id[x]);
        x=fa[top[x]];
    }
    if (d[x]>d[y]) swap(x,y);
    updata(1,1,n,id[x],id[y]);
}
int main(){
    scc(n,m); scc(rt,mod);
    for (int i=1;i<=n;i++) sc(w[i]);
    for (int i=1,x,y;i<n;i++){
        scc(x,y);
        a[x].pb(y),a[y].pb(x);
    }
    dfs1(rt,-1);
    dfs2(rt,rt);
    for (int i=1;i<=n;i++)
        tg=w[i],change(i,i);          //初始值
    for (int i=1,op,x,y;i<=m;i++){
        sc(op);
        if (op==1){
            sccc(x,y,tg);             //x->y路径上加tg
            change(x,y);
        }else if (op==2){
            scc(x,y);                  //询问x->y路径上的和
            printf("%d\n",sum(x,y));
        }else if (op==3){
            scc(x,tg);                  //x为根的子树加tg
            updata(1,1,n,id[x],id[x]+sz[x]-1);
        }else{
            sc(x);                       //询问x为根子树上的和
            printf("%d\n",query(1,1,n,id[x],id[x]+sz[x]-1));
        }
    }
}

标签：剖分,int,top,scc,树链,son,id,模板,define
来源： https://blog.csdn.net/nudt_spy/article/details/91307152

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。