【树链剖分+差分】CC_DGCD Dynamic GCD

【题目】
Codechef
给定一棵 n n n个点的点权树, m m m次操作

  • 询问一条路径上所有数的 g c d gcd gcd
  • 将一段路径的点权加上 d d d

【解题思路】
首先考虑序列上的问题。

我们处理 g c d gcd gcd有一个辗转相减法,即 g c d ( a , b ) = g c d ( b − a , b ) gcd(a,b)=gcd(b-a,b) gcd(a,b)=gcd(ba,b)。于是我们维护一个原序列和一个差分序列,其中差分序列用线段树维护区间的 g c d gcd gcd。那么每次询问操作就是一个区间 g c d gcd gcd和开头的数做 g c d gcd gcd了,修改操作就是单点修改。

每次询问和修改的复杂度都是 O ( log ⁡ n ) O(\log n) O(logn)(因为 g c d gcd gcd一共最多改变 O ( log ⁡ n ) O(\log n) O(logn)次,虽然说其实是值域了)。

丢到树上就套一个树剖就行了,也即将 O ( log ⁡ n ) O(\log n) O(logn)个数字求一个 g c d gcd gcd

复杂度 O ( n log ⁡ 2 n ) O(n\log ^2 n) O(nlog2n)

【参考代码】

#include<bits/stdc++.h>
using namespace std;

const int N=5e4+10;
int n,a[N],w[N];

namespace IO
{
	int read()
	{
		int ret=0;char c=getchar();
		while(!isdigit(c)) c=getchar();
		while(isdigit(c)) ret=ret*10+(c^48),c=getchar();
		return ret;
	}
	void write(int x){if(x>9)write(x/10);putchar(x%10^48);}
	void writeln(int x){write(x);putchar('\n');}
}
using namespace IO;

namespace Segment
{
#define ls (x<<1)
#define rs (x<<1|1)
	int gcd(int x,int y){return y?gcd(y,x%y):x;}
	struct Segment_Tree
	{
		int d[N<<2],tar[N<<2];
		void pushup(int x){d[x]=gcd(d[ls],d[rs]);}
		void build(int x,int l,int r)
		{
			if(l==r){d[x]=w[l]-w[l-1];tar[x]=w[l];return;}
			int mid=(l+r)>>1;
			build(ls,l,mid);build(rs,mid+1,r);
			pushup(x);
		}
		int querynum(int x,int l,int r,int p)
		{
			if(l==r) return tar[x];
			int mid=(l+r)>>1;
			if(p<=mid) return tar[x]+querynum(ls,l,mid,p);
			else return tar[x]+querynum(rs,mid+1,r,p);
		}
		int querygcd(int x,int l,int r,int L,int R)
		{
			if(L>R) return 0;
			if(L<=l && r<=R) return d[x];
			int mid=(l+r)>>1,res=0;
			if(L<=mid) res=gcd(res,querygcd(ls,l,mid,L,R));
			if(R>mid) res=gcd(res,querygcd(rs,mid+1,r,L,R));
			return res;
		}
		void updatenum(int x,int l,int r,int L,int R,int v)
		{
			if(L<=l && r<=R){tar[x]+=v;return;}
			int mid=(l+r)>>1;
			if(L<=mid) updatenum(ls,l,mid,L,R,v);
			if(R>mid) updatenum(rs,mid+1,r,L,R,v);
		}
		void updategcd(int x,int l,int r,int p,int v)
		{
			if(l==r){d[x]+=v;return;}
			int mid=(l+r)>>1;
			if(p<=mid) updategcd(ls,l,mid,p,v);
			else updategcd(rs,mid+1,r,p,v);
			pushup(x);
		}
	}tr;
#undef ls
#undef rs
}
using namespace Segment;

namespace Tree
{
	int tot,ind;
	int head[N],siz[N],fa[N],son[N],pos[N],top[N],dep[N];
	struct Tway{int v,nex;}e[N<<1];
	void add(int u,int v)
	{
		e[++tot]=(Tway){v,head[u]};head[u]=tot;
		e[++tot]=(Tway){u,head[v]};head[v]=tot;
	}
	void dfs1(int x)
	{
		siz[x]=1;
		for(int i=head[x];i;i=e[i].nex)
		{
			int v=e[i].v;
			if(v==fa[x]) continue;
			fa[v]=x;dep[v]=dep[x]+1;dfs1(v);siz[x]+=siz[v];
			if(siz[v]>siz[son[x]]) son[x]=v;
		}
	}
	void dfs2(int x,int tp)
	{
		pos[x]=++ind;top[x]=tp;w[ind]=a[x];
		if(son[x]) dfs2(son[x],tp);
		for(int i=head[x];i;i=e[i].nex)
		{
			int v=e[i].v;
			if(v==fa[x] || v==son[x]) continue;
			dfs2(v,v);  
		}
	}
	void update(int x,int y,int v)
	{
		for(;top[x]^top[y];x=fa[top[x]])
		{
			if(dep[top[x]]<dep[top[y]]) swap(x,y);
			tr.updatenum(1,1,n,pos[top[x]],pos[x],v);
			tr.updategcd(1,1,n,pos[top[x]],v);tr.updategcd(1,1,n,pos[x]+1,-v);
		}
		if(dep[x]>dep[y]) swap(x,y);
		tr.updatenum(1,1,n,pos[x],pos[y],v);
		tr.updategcd(1,1,n,pos[x],v);tr.updategcd(1,1,n,pos[y]+1,-v);
	}
	int query(int x,int y)
	{
		int res=0;
		for(;top[x]^top[y];x=fa[top[x]])
		{

			if(dep[top[x]]<dep[top[y]]) swap(x,y);
			res=gcd(res,tr.querynum(1,1,n,pos[top[x]]));
			res=gcd(res,tr.querygcd(1,1,n,pos[top[x]]+1,pos[x]));
		}

		if(dep[x]>dep[y]) swap(x,y);
		res=gcd(res,tr.querynum(1,1,n,pos[x]));
		res=gcd(res,tr.querygcd(1,1,n,pos[x]+1,pos[y]));
		return abs(res);
	}
}
using namespace Tree;

namespace DreamLolita
{
	int Q;
	void solution()
	{
		n=read();
		for(int i=1;i<n;++i) add(read()+1,read()+1);
		for(int i=1;i<=n;++i) a[i]=read();
		dfs1(1);dfs2(1,1);tr.build(1,1,n);
		Q=read();
		while(Q--)
		{
			char ch[2];scanf("%s",ch);
			int x=read()+1,y=read()+1,d;
			if(ch[0]=='F') printf("%d\n",query(x,y));
			else d=read(),update(x,y,d);
		}
	}
}

int main()
{
#ifndef ONLINE_JUDGE
	freopen("DGCD.in","r",stdin);
	freopen("DGCD.out","w",stdout);
#endif
	DreamLolita::solution();
	return 0;
}

【总结】
最近树剖写得真多。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值