2016计蒜之道初赛第一场--B青云的机房组网方案（简单）【floyd】

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

输入格式

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

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

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

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

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

输出格式

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

样例输入

4
1 2 3 4
1 2
2 3
3 4

样例输出

8

提示信息

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

$(1,2)$：$1$

$(1,3)$：$2$

$(1,4)$：$3$

$(2,3)$：$1$

$(3,4)$：$1$

所以，所有初始值互质的机房对的传输时间和为 $1+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);
    }
}