【模板】树链剖分

模板题

#include
#include
#include
using namespace std;
#define lch i<<1
#define rch i<<1|1
#define N 30005
struct mzls
{
	int to,nt;
}a[N*2];
struct Tree
{
	int l,r,mx,sum;
}t[N*4];
int n,head[N],dep[N],top[N],siz[N],son[N],fat[N],id[N];
int cnt,tot;
inline void add(int x,int y)
{
	a[cnt].to=y,a[cnt].nt=head[x],head[x]=cnt++;
}
inline void dfs(int x,int fa,int d)
{
	fat[x]=fa,dep[x]=d,siz[x]=1,son[x]=0;
	int s1=0;
	for(int i=head[x];i!=-1;i=a[i].nt)
		if(a[i].to!=fa)
		{
			int v=a[i].to;
			dfs(v,x,d+1);
			siz[x]+=siz[v];
			if(siz[v]>siz[s1])
				s1=v;
		}
	son[x]=s1;
}
inline void dfs1(int x,int tp)
{
	id[x]=++tot,top[x]=tp;
	if(son[x])
		dfs1(son[x],tp);
	for(int i=head[x];i!=-1;i=a[i].nt)
		if(a[i].to!=fat[x]&&a[i].to!=son[x])
			dfs1(a[i].to,a[i].to);
}
inline void Build(int i,int l,int r)
{
	t[i].l=l,t[i].r=r;
	if(l==r)
		return;
	int mid=(l+r)>>1;
	Build(lch,l,mid);
	Build(rch,mid+1,r);
}
inline void Up(int i)
{
	t[i].mx=max(t[lch].mx,t[rch].mx);
	t[i].sum=t[lch].sum+t[rch].sum;
}
inline void modify(int i,int d1,int ml)
{
	if(t[i].l==t[i].r)
	{
		t[i].mx=t[i].sum=ml;
		return;
	}
	int mid=(t[i].l+t[i].r)>>1;
	if(d1<=mid)
		modify(lch,d1,ml);
	else
		modify(rch,d1,ml);
	Up(i);
}
inline int q1(int i,int l,int r)
{
	if(t[i].rr)
		return -0x3f3f3f3f;
	if(t[i].r<=r&&t[i].l>=l)
		return t[i].mx;
	return max(q1(lch,l,r),q1(rch,l,r));
}
inline int q2(int i,int l,int r)
{
	if(t[i].rr)
		return 0;
	if(t[i].r<=r&&t[i].l>=l)
		return t[i].sum;
	return q2(lch,l,r)+q2(rch,l,r);
}
inline int Q1(int u,int v)
{
	int t1=top[u],t2=top[v];
	int ans=-0x3f3f3f3f;
	while(t1!=t2)
	{
		if(dep[t1]

 

你可能感兴趣的:(算法模板,树链剖分)