2016计蒜之道初赛第一场--B青云的机房组网方案（简单）【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。

思路：n最大500，直接floyd+互质数判断暴力搞就行，不需要任何优化。

AC代码：

#include<stdio.h>
#include<string.h>
#include<iostream>
using namespace std;
int map[505][505];
int a[500];
int huzhi(int x,int y)
{
for(int i=2;i<=x&&i<=y;i++)
{
if(x%i==0&&y%i==0)
{
return 0;
}
}
return 1;
}
int main()
{
int n;
while(~scanf("%d",&n))
{
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
map[i][j]=0x3f3f3f3f;
}
}
for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
}
for(int i=0;i<n-1;i++)
{
int x,y;
scanf("%d%d",&x,&y);
x--;y--;
map[x][y]=map[y][x]=1;
}
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
for(int k=0;k<n;k++)
{
map[j][k]=min(map[j][k],map[j][i]+map[i][k]);
}
}
}
int output=0;
for(int i=0;i<n;i++)
{
for(int j=i+1;j<n;j++)
{
if(huzhi(a[i],a[j]))
{

output+=map[i][j];
}
}
}
printf("%d\n",output);
}
}