• 洛谷 [P3384] 树链剖分 模版


    支持各种数据结构上树,注意取膜.

    #include <iostream>
    #include <cstring>
    #include <algorithm>
    #include <cstdlib>
    #include <cstdio>
    #include <queue>
    #define lson l,mid,rt<<1
    #define rson mid+1,r,rt<<1|1
    #define LL long long 
    using namespace std;
    const int MAXN=100005;
    int init(){
    	int rv=0,fh=1;
    	char c=getchar();
    	while(c<'0'||c>'9'){
    		if(c=='-') fh=-1;
    		c=getchar();
    	}
    	while(c>='0'&&c<='9'){
    		rv=(rv<<1)+(rv<<3)+c-'0';
    		c=getchar();
    	}
    	return fh*rv;
    }
    int ind,head[MAXN],dep[MAXN],son[MAXN],nume,n,m,MOD,root,id[MAXN],siz[MAXN],wei[MAXN],wein[MAXN],top[MAXN],fa[MAXN];
    struct edge{
    	int to,nxt;
    }e[MAXN<<1];
    void adde(int from,int to){
    	e[++nume].to=to;
    	e[nume].nxt=head[from];
    	head[from]=nume;
    }
    void dfs1(int u,int rt){
    	dep[u]=dep[rt]+1;
    	fa[u]=rt;
    	siz[u]=1;
    	int ma=0;
    	for(int i=head[u];i;i=e[i].nxt){
    		int v=e[i].to;
    		if(v==rt) continue;
    		dfs1(v,u);
    		siz[u]+=siz[v];
    		if(siz[v]>ma){
    			ma=siz[v];
    			son[u]=v;
    		}
    	}
    }
    void dfs2(int u,int topf){
    	id[u]=++ind;
    	top[u]=topf;
    	wein[ind]=wei[u];
    	if(!son[u]) return ;
    	dfs2(son[u],topf);
    	for(int i=head[u];i;i=e[i].nxt){
    		int v=e[i].to;
    		if(v==fa[u]||v==son[u]) continue;
    		dfs2(v,v);
    	}
    }
    struct SGT{
    	LL sum[MAXN<<2],tag[MAXN<<2];
    	void PushUp(int rt){
    		sum[rt]=(sum[rt<<1]+sum[rt<<1|1])%MOD;
    	}
    	void build(int l,int r,int rt){
    		if(l==r) {
    			sum[rt]=wein[l];
    			return;
    		}
    		int mid=(l+r)>>1;
    		if(l<=mid) build(lson);
    		if(mid<r) build(rson);
    		PushUp(rt);
    	}
    	void PushDown(int rt,int m){
    		if(tag[rt]){
    			(tag[rt<<1]+=tag[rt])%=MOD;
    			(tag[rt<<1|1]+=tag[rt])%=MOD;
    			(sum[rt<<1]+=tag[rt]*(m-(m>>1)))%=MOD;
    			(sum[rt<<1|1]+=tag[rt]*(m>>1))%=MOD;
    			tag[rt]=0;
    		}
    	}
    	void Update(int L,int R,int add,int l,int r,int rt){
    		if(L<=l&&r<=R){
    			(tag[rt]+=add)%=MOD;
    			(sum[rt]+=(LL)add*(r-l+1))%=MOD;
    			return;
    		}
    		int mid=(l+r)>>1;
    		PushDown(rt,r-l+1);
    		if(L<=mid) Update(L,R,add,lson);
    		if(mid<R) Update(L,R,add,rson);
    		PushUp(rt);
    	}
    	int query(int L,int R,int l,int r,int rt){
    		if(L<=l&&r<=R){
    			return sum[rt];
    		}
    		LL ret=0;
    		int mid=(l+r)>>1;
    		PushDown(rt,r-l+1);
    		if(L<=mid) ret+=query(L,R,lson);
    		if(mid<R) ret+=query(L,R,rson);
    		return ret%MOD;
    	}
    }sgt;
    void UpdateRange(int u,int v,int add){
    	add%=MOD;
    	while(top[u]!=top[v]){
    		if(dep[top[u]]<dep[top[v]]) swap(u,v);
    		sgt.Update(id[top[u]],id[u],add,1,n,1);
    		u=fa[top[u]];
    	}
    	if(dep[u]>dep[v]) swap(u,v);
    	sgt.Update(id[u],id[v],add,1,n,1);
    }
    int QueryRange(int u,int v){
    	int ans=0;
    	while(top[u]!=top[v]){
    		if(dep[top[u]]<dep[top[v]]) swap(u,v);
    		(ans+=sgt.query(id[top[u]],id[u],1,n,1))%=MOD;
    		u=fa[top[u]];
    	}
    	if(dep[u]>dep[v]) swap(u,v);
    	(ans+=sgt.query(id[u],id[v],1,n,1))%=MOD;
    	return ans%MOD;
    }
    int main(){
    	n=init();m=init();root=init();MOD=init();
    	for(int i=1;i<=n;i++) wei[i]=init();
    	for(int i=1;i<n;i++){
    		int u=init(),v=init();
    		adde(u,v);adde(v,u);
    	}
    	dep[root]=1;
    	dfs1(root,0);
    	dfs2(root,root);
    	sgt.build(1,n,1);
    	for(int i=1;i<=m;i++){
    		int t=init();
    		if(t==1){
    			int u=init(),v=init(),ttt=init();
    			UpdateRange(u,v,ttt);
    		}else if(t==2){
    			int u=init(),v=init();
    			printf("%d
    ",QueryRange(u,v));
    		}else if(t==3){
    			int tt=init(),ttt=init();
    			sgt.Update(id[tt],id[tt]+siz[tt]-1,ttt%MOD,1,n,1);
    		}else {
    			int ttt=init();
    			printf("%d
    ",sgt.query(id[ttt],id[ttt]+siz[ttt]-1,1,n,1));
    		}
    	}
    	return 0;
    }
    
  • 相关阅读:
    sqlserver 跟踪标志
    解决ORA-00338,ORA-00312
    oracle SQL性能分析
    高潜力人士和员工
    pymysqlreplication
    Python3操作Excel(写入)
    CentOS7.4 源码安装MySQL8.0
    MySql 时间操作实例
    python+eclipse+pydev开发环境搭建
    MySQL表结构同步工具 mysql-schema-sync
  • 原文地址:https://www.cnblogs.com/Mr-WolframsMgcBox/p/8407012.html
Copyright © 2020-2023  润新知