树链剖分题目汇总+板子

辣鸡的我终于开始学树链剖分了,而ly聚聚早都会了QAQ.....

首先是一个板子题,就是板子

洛谷-P3384-【模板】树链剖分

题目链接:https://www.luogu.org/problemnew/show/P3384

题目大意:四种操作,板子。

思路:板子。线段树维护分出来的链。

ACCode(板子):

//#pragma comment(linker, "/STACK:1024000000,1024000000")
  
#include<stdio.h>
#include<string.h> 
#include<math.h> 
   
#include<map>  
#include<set>
#include<deque> 
#include<queue> 
#include<stack> 
#include<bitset>
#include<string> 
#include<fstream>
#include<iostream> 
#include<algorithm> 
using namespace std; 
  
#define ll long long 
#define Pair pair<int,int>
//#define max(a,b) (a)>(b)?(a):(b)
//#define min(a,b) (a)<(b)?(a):(b)
#define clean(a,b) memset(a,b,sizeof(a))// ??
//std::ios::sync_with_stdio(false);
//  register
const int MAXN=1e5+10;
const int INF32=0x3f3f3f3f;
const ll INF64=0x3f3f3f3f3f3f3f3f;
const ll mod=1e9+7;
const double PI=acos(-1.0);
const double EPS=1.0e-8;

class Segment{
	int P;
	int Tree[MAXN<<2],Lazy[MAXN<<2],Size[MAXN<<2];
	
	void PushDown(int rt){
		if(Lazy[rt]==0) return ;
		Tree[rt<<1]=(Tree[rt<<1]+Size[rt<<1]*Lazy[rt]+P)%P;
		Tree[rt<<1|1]=(Tree[rt<<1|1]+Size[rt<<1|1]*Lazy[rt]+P)%P;
		
		Lazy[rt<<1]=(Lazy[rt<<1]+Lazy[rt]+P)%P;
		Lazy[rt<<1|1]=(Lazy[rt<<1|1]+Lazy[rt]+P)%P;
		Lazy[rt]=0;
	}
	public : void SetMod(int Mod){
		P=Mod;
	}
	public : void Build(int l,int r,int rt,int a[]){
		Size[rt]=r-l+1;
//		printf("rt=%d l=%d r=%d Tree[rt]=%d\n",rt,l,r,Tree[rt]);
		if(l==r){
			Tree[rt]=a[l];
			return ;
		}int mid=(l+r)>>1;
		Build(l,mid,rt<<1,a);Build(mid+1,r,rt<<1|1,a);
		Tree[rt]=(Tree[rt<<1]+Tree[rt<<1|1]+P)%P;
	}
	public : void Update(int ql,int qr,int val,int l,int r,int rt){
		if(ql<=l&&r<=qr){
			Tree[rt]+=Size[rt]*val;
			Lazy[rt]+=val;
			return ;
		}int mid=(l+r)>>1;
		PushDown(rt);
		if(ql<=mid) Update(ql,qr,val,l,mid,rt<<1);
		if(qr>mid) Update(ql,qr,val,mid+1,r,rt<<1|1);
		Tree[rt]=(Tree[rt<<1]+Tree[rt<<1|1]+P)%P;
	}
	public : int Query(int ql,int qr,int l,int r,int rt){
		if(ql<=l&&r<=qr) return Tree[rt];
		PushDown(rt);
		int mid=(l+r)>>1;
		int ans=0;
		if(ql<=mid) ans=(ans+Query(ql,qr,l,mid,rt<<1)+P)%P;
		if(qr>mid) ans=(ans+Query(ql,qr,mid+1,r,rt<<1|1)+P)%P;
		return ans;
	}
	public : void Show(int l,int r,int rt){
		printf("rt=%d l=%d r=%d Tree[rt]=%d\n",rt,l,r,Tree[rt]);
		if(l==r) return ;
		int mid=(l+r)>>1;
		Show(l,mid,rt<<1);Show(mid+1,r,rt<<1|1);
	}
};
struct Node1{
	int v,val,nxt;
	Node1(int _v=0,int _val=0,int _nxt=0){
		v=_v;val=_val;nxt=_nxt;
	}
};
Segment Seg;
Node1 Edge[MAXN<<2];
int Head[MAXN],Ecnt;
int Deep[MAXN],Fa[MAXN],Size[MAXN],Son[MAXN];
int Idx[MAXN],Icnt;//重新标号 
int Top[MAXN];
int A[MAXN],B[MAXN];
int n,m,R,P;

void Intt(){
	clean(Head,-1);Ecnt=0;
	clean(Deep,0);clean(Fa,-1);clean(Size,0);clean(Son,-1);
	Icnt=0;
}
void AddEdge(int u,int v,int val){
	Edge[Ecnt]=Node1(v,val,Head[u]);
	Head[u]=Ecnt++;
}
int DFS1(int u,int fa,int dep){
	Deep[u]=dep;
	Fa[u]=fa;
	Size[u]=1;
	int maxson=-1;
	for(int i=Head[u];i+1;i=Edge[i].nxt){
		int temp=Edge[i].v;
		if(temp==fa) continue;
		Size[u]+=DFS1(temp,u,dep+1);
		if(Size[temp]>maxson){
			maxson=Size[temp];
			Son[u]=temp;
		}
	}return Size[u];
}
void DFS2(int u,int topfa){
	Idx[u]=++Icnt;
//	cout<<"u,Idx[u]"<<u<<" "<<Idx[u]<<endl;
	Top[u]=topfa;
	B[Idx[u]]=A[u];
	if(Son[u]==-1) return ;
	DFS2(Son[u],topfa);
	for(int i=Head[u];i+1;i=Edge[i].nxt){
		int temp=Edge[i].v;
		if(Idx[temp]==0){
			DFS2(temp,temp);
		}
	}
}
void UpdatePath(int l,int r,int val){
	while(Top[l]!=Top[r]){
		if(Deep[Top[l]]<Deep[Top[r]]) swap(l,r);
		Seg.Update(Idx[Top[l]],Idx[l],val,1,n,1);
		l=Fa[Top[l]];
	}
	if(Deep[l]>Deep[r]) swap(l,r);
	Seg.Update(Idx[l],Idx[r],val,1,n,1);
}
int QueryPath(int l,int r){
	int ans=0;
	while(Top[l]!=Top[r]){
		if(Deep[Top[l]]<Deep[Top[r]]) swap(l,r);
		ans=(ans+Seg.Query(Idx[Top[l]],Idx[l],1,n,1)+P)%P;
		l=Fa[Top[l]];
	}
	if(Deep[l]>Deep[r]) swap(l,r);
	ans=(ans+Seg.Query(Idx[l],Idx[r],1,n,1)+P)%P;
	return ans;
}
void UpdateSubTree(int rt,int val){
	int l=Idx[rt],r=Idx[rt]+Size[rt]-1;
	Seg.Update(l,r,val,1,n,1);
}
int QuerySubTree(int rt){
	int l=Idx[rt],r=Idx[rt]+Size[rt]-1;
	int ans=Seg.Query(l,r,1,n,1);
	return ans;
}
int main(){
	scanf("%d%d%d%d",&n,&m,&R,&P);
	Intt();Seg.SetMod(P);
	for(int i=1;i<=n;++i){
		scanf("%d",&A[i]);
	}
	for(int i=1;i<n;++i){
		int x,y;scanf("%d%d",&x,&y);
		AddEdge(x,y,1);AddEdge(y,x,1);
	}
	DFS1(R,-1,1);
//	cout<<"DFS2:"<<endl;
	DFS2(R,R);
//	cout<<"Icnt: "<<Icnt<<endl;
//	for(int i=1;i<=n;++i){
//		cout<<"i,Idx[i],Size[i]: "<<i<<" "<<Idx[i]<<" "<<Size[i]<<endl;
//	}
	Seg.Build(1,n,1,B);
//	Seg.Show(1,n,1);
	for(int i=1;i<=m;++i){
		int oper;scanf("%d",&oper);
		if(oper==1){//x,y,z,x节点到y节点上路径的所有节点+z 
			int x,y,z;scanf("%d%d%d",&x,&y,&z);
			UpdatePath(x,y,z);
		}
		else if(oper==2){//求x,y节点的 路径之和 
			int x,y;scanf("%d%d",&x,&y);
			printf("%d\n",QueryPath(x,y));
		}
		else if(oper==3){//x节点的子树都+z 
			int x,z;scanf("%d%d",&x,&z);
			UpdateSubTree(x,z);
		}
		else{//求x节点的子树和 
			int x;scanf("%d",&x);
			printf("%d\n",QuerySubTree(x));
		}
	}
	
	
}

好了,板子已经有了,剩下的就是一些练习了:

洛谷-P3379 【模板】最近公共祖先(LCA)(强行用树剖写LCA)

题目链接:https://www.luogu.org/problemnew/show/P3379

状态:已AC,题解:https://blog.csdn.net/qq_40482358/article/details/89677947

洛谷-P2590 [ZJOI2008]树的统计

题目链接:https://www.luogu.org/problemnew/show/P2590

状态:已AC,题解:https://blog.csdn.net/qq_40482358/article/details/89679241

洛谷-P3178 [HAOI2015]树上操作

题目链接:https://www.luogu.org/problemnew/show/P3178

状态:已AC,题解:https://blog.csdn.net/qq_40482358/article/details/89680488

洛谷P3038 [USACO11DEC]牧草种植Grass Planting

题目链接:  https://www.luogu.org/problemnew/show/P3038

状态:已AC,题解:https://blog.csdn.net/qq_40482358/article/details/89853548

洛谷-P2146 [NOI2015]软件包管理器

题目链接: https://www.luogu.org/problemnew/show/P2146

状态:已AC,题解:https://blog.csdn.net/qq_40482358/article/details/90176500

洛谷-P2486 [SDOI2011]染色

题目链接:https://www.luogu.org/problemnew/show/P2486

状态:已AC,题解:https://blog.csdn.net/qq_40482358/article/details/90208349

-----树链剖分专题暂时告一段落了,虽然写的都是些水题,但我也初步了解树链剖分了,以后碰到好的题目再加进来吧。---------

Update:2019/5/15/ 13:54.

写之前的专题的时候碰到了几道树剖的题:虽然不算难,但一起总结起来吧。

POJ-2763-Housewife Wind

题目链接:http://poj.org/problem?id=2763

题解:https://blog.csdn.net/qq_40482358/article/details/90234518

 

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值