CF990G

题意

https://codeforces.com/contest/990/problem/G

---

思考

在200000以内，因数个数最多的数位166320，共有160个因数。可以知道，从一个节点向下走最多只会有160种取值。

记集合f[u]为从u节点向下走可以取得的所有值及其个数，暴力转移即可。

至于合并，博主写了平方.......但这题想造出极端数据也是困难的。

最后注意空间。

---

代码

 #include<bits/stdc++.h>
 using namespace std;
 typedef long long int ll;
 const int maxn=2E5+;
 const ll base=2E5+;
 int n,val[maxn];
 int head[maxn*],size;
 ll ans[maxn],wait[maxn];
 bool vis[maxn];
 inline int gcd(int x,int y)
 {
     return x%y==?y:gcd(y,x%y);
 }
 struct edge
 {
     int to,next;
 }E[maxn*];
 inline void add(int u,int v)
 {
     E[++size].to=v;
     E[size].next=head[u];
     head[u]=size;
 }
 struct note
 {
     int x,c;
     note(int a=,int b=)
     {
         x=a,c=b;
     }
 };
 queue<note>T[maxn];
 vector<note>t;
 void dfs(int u,int F)
 {
     vector<int>what;
     for(int i=head[u];i;i=E[i].next)
     {
         int v=E[i].to;
         if(v==F)
             continue;
         dfs(v,u);
     }
     for(int i=head[u];i;i=E[i].next)
     {
         int v=E[i].to;
         if(v==F)
             continue;
         t.clear();
         while(!T[v].empty())
         {
             t.push_back(T[v].front());
             T[v].pop();
         }
         for(int j=;j<what.size();++j)
         {
             int x=what[j];
             for(int k=;k<t.size();++k)
                 ans[gcd(x,t[k].x)]+=wait[x]*t[k].c;
         }
         for(int j=;j<t.size();++j)
         {
             int x=gcd(val[u],t[j].x);
             if(!vis[x])
             {
                 vis[x]=;
                 wait[x]=t[j].c;
                 what.push_back(x);
             }
             else
                 wait[x]+=t[j].c;
         }
     }
     if(!vis[val[u]])
     {
         vis[val[u]]=;
         wait[val[u]]=;
         what.push_back(val[u]);
     }
     else
         ++wait[val[u]];
     for(int i=;i<what.size();++i)
     {
         ll x=what[i];
         ans[x]+=wait[x];
         T[u].push(note(x,wait[x]));
         vis[x]=wait[x]=;
     }
 }
 int get(int x)
 {
     int sum=;
     for(int i=;i*i<=x;++i)
         if(x%i==)
             sum+=;
     return sum;
 }
 int main()
 {
     ios::sync_with_stdio(false);
     cin>>n;
     for(int i=;i<=n;++i)
         cin>>val[i];
     for(int i=;i<=n-;++i)
     {
         int x,y;
         cin>>x>>y;
         add(x,y);
         add(y,x);
     }
     dfs(,);
     for(int i=;i<=;++i)
         if(ans[i])
             cout<<i<<" "<<ans[i]<<endl;
     return ;
 }