HDU - 6430 TeaTree

题目链接:HDU - 6430


水题。

显然一个数字的因子不会太多。暴力求出每个数字的因子,然后从子树线段树合并即可。每次合并的时候更新答案。


AC代码:

#pragma GCC optimize("-Ofast","-funroll-all-loops")
#include
//#define int long long
using namespace std;
const int N=1e5+10,M=N*400;
int n,up,rt[N],lc[M],rc[M],sum[M],res[N],a[N],cnt,flag;
vector<int> g[N],v[N];
void get(int n){for(int i=1;i<=n;i++)	for(int j=i;j<=n;j+=i)	v[j].push_back(i);}
void change(int &p,int l,int r,int x){
	if(!p)	p=++cnt;
	if(l==r){sum[p]++;	return ;}	int mid=l+r>>1;
	if(x<=mid)	change(lc[p],l,mid,x);
	else	change(rc[p],mid+1,r,x);
}
int merge(int x,int y,int l,int r){
	if(!x||!y)	return x+y;
	if(l==r){if(sum[x]&&sum[y])	res[flag]=max(res[flag],l);		sum[x]+=sum[y]; return x;}
	int mid=l+r>>1;
	lc[x]=merge(lc[x],lc[y],l,mid);
	rc[x]=merge(rc[x],rc[y],mid+1,r);
	return x;
}
void dfs(int x){
	for(int i=0;i<v[a[x]].size();i++)	change(rt[x],1,up,v[a[x]][i]);
	for(auto to:g[x]){dfs(to);	flag=x;	rt[x]=merge(rt[x],rt[to],1,up);}
}
signed main(){
	get(1e5);	cin>>n;
	for(int i=2,x;i<=n;i++)	scanf("%d",&x),g[x].push_back(i);
	for(int i=1;i<=n;i++)	scanf("%d",&a[i]),up=max(up,a[i]);
	memset(res,-1,sizeof res);	dfs(1);	
	for(int i=1;i<=n;i++)	printf("%d\n",res[i]);
	return 0;
}

你可能感兴趣的:(HDU,线段树)