floyd算法 青云的机房组网方案（简单）

青云的机房组网方案（简单）

青云现在要将 nn 个机房连成一个互相连通的网络。工程师小王设计出一个方案：通过在 nn 个机房之间铺设 n-1n−1 条双向的光纤，将所有的机房连接。可以假设数据在两个机房之间的光纤传输需要 11 单位时间。每个机房 ii 有一个初始值 a_ia​i​​，当两个机房的初始值之间互质时，我们认为这两个机房之间的传输性能是非常重要的。请帮小王计算出所有数值互质的机房对之间的传输时间之和。

输入格式

第一行输入一个正整数 nn，第二行输入 nn 个正整数 a_1...a_na​1​​...a​n​​，表示 nn 个机房的初始值。

接下来输入 n-1n−1 行，每行输入两个数 a,ba,b，表示机房 aa 和机房 bb 之间有一条双向网络管道。

对于简单版本：n \leq 500n≤500，1 \leq a_i \leq 501≤a​i​​≤50；

对于中等版本：n \leq 10000n≤10000， 1 \leq a_i \leq 5001≤a​i​​≤500；

对于困难版本：n \leq 100000n≤100000，a_i \leq 100000a​i​​≤100000。

输出格式

输出一行，表示所有初始值互质的机房对的传输时间和。

样例输入

4
1 2 3 4
1 2
2 3
3 4

样例输出

8

提示信息

对于第一组样例，每一组初始值互质的机房对的传输时间如下：

(1,2)(1,2)：11

(1,3)(1,3)：22

(1,4)(1,4)：33

(2,3)(2,3)：11

(3,4)(3,4)：11

所以，所有初始值互质的机房对的传输时间和为 1+2+3+1+1=81+2+3+1+1=8

xyd给我报上这个比赛，结果他自己都不做了，我就做出一个

 /*floyd算法：直接做就可以了，暴力统计结果*/
 #include<iostream>
 using namespace std;
 #define N 510
 long long jz[N][N];
 #include<cstdio>
 #include<cstring>
 int a[N],n,x,y;
 int gcd(int b,int c)
 {
     if(!c)
         return b;
     return gcd(c,b%c);
 }
 void input()
 {
     scanf("%d",&n);
     memset(jz,,sizeof(jz));
     for(int i=;i<=n;++i)
     {
         scanf("%d",&a[i]);
     }
     for(int i=;i<=n-;++i)
     {
         scanf("%d%d",&x,&y);
         jz[x][y]=;jz[y][x]=;
     }
 }
 void floyed()
 {
     for(int k=;k<=n;++k)
       for(int i=;i<=n;++i)
         for(int j=;j<=n;++j)
         jz[i][j]=min(jz[i][j],jz[i][k]+jz[k][j]);
 }
 int main()
 {
     input();
     floyed();
     long long ans=;
     for(int i=;i<=n;++i)
       for(int j=i+;j<=n;++j)
       {
           if(gcd(a[i],a[j])==)
           {
               ans+=jz[i][j];
           }
       }
     cout<<ans<<endl;
     return ;
 }