hdu 6430 TeaTree 线段树合并

hdu 6430

多校十第五题，我是从丿残念灬这位大佬学的，从来没写过线段树合并，第一次发现还有这操作，学到了学到了，这个题求任意点 i 和以其为根的子树的所有点 j 的max( gcd  (v[ i ], v[ j ]) )，写法是建立n个线段树，每个线段树保存的是该点权值的所有因数，首先将1e5个数的所有因数打表，然后再将n个点的权值一一建线段树，复杂度应该是n*logn*logn，然后走dfs，从底部向上合并节点，每次都把儿子节点合并到父节点，所以每个点只被合并一次，合并的复杂度大概也是之前的数量级，在合并的过程中求出父节点的最大的gcd(fa,son)，具体怎么合并，看代码更容易理解。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=1e5+10;
int rt[maxn],ls[maxn*400],rs[maxn*400],sum[maxn*400];
vector<int>G[maxn],fac[maxn];
int cnt=0,sz=1e5,ans[maxn];
void init()
{
	for(int i=1;i<maxn;i++)
	for(int j=i;j<maxn;j+=i)
	fac[j].push_back(i);//将每个数的所有因数保存起来 
}
void pushup(int o)
{
	sum[o]=max(sum[ls[o]],sum[rs[o]]);
}
void update(int k,int l,int r,int &o)
{
	if(!o)o=++cnt;
	if(l==r)
	{
		sum[o]=k;
		return;
	}
	int m=(l+r)/2;
	if(k<=m)update(k,l,m,ls[o]);
	else update(k,m+1,r,rs[o]);
	pushup(o); 
}
int merge(int o,int pre,int& v)
{
	if(!o||!pre)return o|pre;//将父线段树的因数和儿子线段树因数合并到父线段树， 
	if(sum[o]==sum[pre])v=max(v,sum[o]);
	if(ls[o]|ls[pre])ls[o]=merge(ls[o],ls[pre],v);//父线段树和儿子线段树左节点都是空的，就不用遍历合并 
	if(rs[o]|rs[pre])rs[o]=merge(rs[o],rs[pre],v);
	return o;
}
void dfs(int u)
{
	ans[u]=-1;
	for(int i=0;i<G[u].size();i++)
	{
		int v=G[u][i];
		dfs(v);
		merge(rt[u],rt[v],ans[u]);
	}
}
int main()
{
	int n,v;
	init();
	scanf("%d",&n);
	for(int i=2;i<=n;i++)
	{
		scanf("%d",&v);
		G[v].push_back(i);
	}
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&v);
		for(int j=0;j<fac[v].size();j++)
		update(fac[v][j],1,sz,rt[i]);
	}
	dfs(1);
	for(int i=1;i<=n;i++)
	printf("%d\n",ans[i]);
}