csp模拟 路径【复杂度分析】

传送门

其实就是个复杂度分析和基础数论。

众所周知，gcd变化至少减半，所以最多log次变换。

任何一条路径都可以成为以某个叶子结点为根，的一条从上到下的链。

而叶子最多10个。

所以从每个叶子dfs，暴力维护每个点到根路径上的不同gcd点。然后每次更新即可。

复杂度10个nlogn

#include<bits/stdc++.h>
using namespace std;
#define in read()
inline int in{
	int cnt=0,f=1;char ch=0;
	while(!isdigit(ch)){
		ch=getchar();if(ch=='-')f=-1;
	}
	while(isdigit(ch)){
		cnt=cnt*10+ch-48;
		ch=getchar();
	}return cnt*f;
}
struct node{
	int cnt;
	pair<int,int> a[37];
}a[170003];long long ans;
int n;
int first[170003],nxt[320003],to[320003],tot;
void add(int a,int b){
	nxt[++tot]=first[a];first[a]=tot;to[tot]=b;
}int key[160003],rd[160003],dep[160003];
int gcd(int a,int b){return b?gcd(b,a%b):a;};
void dfs(int u,int faa){
	dep[u]=dep[faa]+1;a[u]=a[faa];a[u].a[++a[u].cnt]=make_pair(u,key[u]);
	for(register int i=1;i<=a[u].cnt;i++){
		a[u].a[i]=make_pair(a[u].a[i].first,gcd(a[u].a[i].second,key[u]));
		ans=max(ans,(long long)a[u].a[i].second*(dep[u]-dep[a[u].a[i].first]+1));
	}int num=0;
	for(register int i=1;i<=a[u].cnt;i++)if(a[u].a[i].second!=a[u].a[i-1].second)
	a[u].a[++num]=a[u].a[i];a[u].cnt=num;
	for(register int i=first[u];i;i=nxt[i]){
		int v=to[i];if(v==faa)continue;dfs(v,u);
	}
}
signed main(){
	n=in;for(register int i=1;i<=n;i++)key[i]=in;for(register int i=1;i<n;i++){
		int x=in;int y=in;add(x,y);add(y,x);rd[x]++;rd[y]++;
	}for(register int i=1;i<=n;i++)if(rd[i]==1)dfs(i,0);cout<<ans;
	return 0;
}