【题目】
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(b−a,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;
}
【总结】
最近树剖写得真多。