HDU-4081-Qin Shi Huang's National Road System （次小生成树）

题目链接：HDU-4081-Qin Shi Huang’s National Road System

题意是给定坐标系上的n个点的坐标和人口，要求这 $n$ 个点之间构成一棵树， $A$ 定义为该边两端点的人口数之和， $B$ 定义为除了该边外其他边的长度。求 $\frac{A}{B}$ 的最大值。

考虑到 $n<=10^3$ ，那么边的数量 $e<10^6$ ，所以枚举每一条边，每一条边的 $A$ 是定值，所以求 $B$ 的最小值。最小生成树的权值为 $cost$ ， 则$B$ 的最小值为

$$cost-G[u][v] \quad (当u,v是构成最小生成树的边)$$
或

$$cost-Max[u][v] \quad (当u,v不是构成最小生成树的边)$$

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1007;
const double INF=1e9+7;
double G[maxn][maxn],Max[maxn][maxn],d[maxn];
int p[maxn],x[maxn],y[maxn],pre[maxn];
bool vis[maxn],used[maxn][maxn];
int n;
double Prim()
{
    double val=0;
    memset(vis,0,sizeof(vis));
    memset(used,0,sizeof(used));
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)   Max[i][j]=0;
    vis[1]=true;
    for(int i=1;i<=n;i++)   d[i]=G[1][i],pre[i]=1;
    for(int i=1;i<n;i++)
    {
        int u;
        double c=INF;
        for(int v=1;v<=n;v++)
        {
            if(!vis[v]&&d[v]<c)
            {
                u=v;
                c=d[v];
            }
        }
        for(int v=1;v<=n;v++)
            if(vis[v])  Max[u][v]=Max[v][u]=max(c,Max[v][pre[u]]);
        vis[u]=true;
        val+=c;
        used[u][pre[u]]=used[pre[u]][u]=true;
        for(int v=1;v<=n;v++)
        {
            if(!vis[v]&&d[v]>G[u][v])   d[v]=G[u][v],pre[v]=u;
        }
    }
    double res=0;
    for(int i=1;i<=n;i++)
        for(int j=i+1;j<=n;j++)
        {
            if(used[i][j])
                res=max(res,(p[i]+p[j])/(val-G[i][j]));
            else
                res=max(res,(p[i]+p[j])/(val-Max[i][j]));
        }
    return res;
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%d%d%d",&x[i],&y[i],&p[i]);
        }
        for(int i=1;i<=n;i++)
        {
            G[i][i]=0;
            for(int j=i+1;j<=n;j++)
                G[i][j]=G[j][i]=sqrt(pow(x[i]-x[j],2)+pow(y[i]-y[j],2));
        }
        printf("%.2f\n",Prim());
    }
    return 0;
}