UVALive 5713 Qin Shi Huang's National Road System秦始皇修路（MST，最小瓶颈路）

  1 #include <bits/stdc++.h>
  2 #include <cstdio>
  3 #include <vector>
  4 #include <iostream>
  5 #include <algorithm>
  6 #define INF 0x7f7f7f7f
  7 #define pii pair<int,int>
  8 #define LL long long
  9 using namespace std;
 10 const int N=1001000;
 11 int a[N], b[N], seq[N];     //求MST用的
 12 int x[N], y[N], p[N];       //所给的坐标及人口数
 13 int pre[N], vis[N], used[N];    //求任意点对最小瓶颈路用的
 14 double w[N], maxcost[1001][1001]; //两点间的最小瓶颈maxcost
 15 vector<int>  vect[N];    //建树时用
 16 vector<int> dfn;         //记录访问过的节点
 17 
 18 int cmp(int a,int b){return w[a]<w[b];}
 19 int find(int x){return pre[x]==x? x: pre[x]=find(pre[x]);}  //并查集
 20 double dis(int a,int b){return sqrt((x[a]-x[b])*(x[a]-x[b])+(y[a]-y[b])*(y[a]-y[b]));}
 21 
 22 
 23 
 24 
 25 void DFS(int x)
 26 {
 27     dfn.push_back(x);   //访问过
 28     vis[x]=1;
 29     for(int i=0; i<vect[x].size(); i++)
 30     {
 31         int t=vect[x][i];
 32         if(!vis[t] )
 33         {
 34             for(int j=0; j<dfn.size(); j++) //对于所有已经访问过的节点
 35             {
 36                 int from=dfn[j];
 37                 maxcost[t][from]=maxcost[from][t]=max(maxcost[from][x], dis(x, t) );//通过x连到t
 38             }
 39             DFS(t);
 40         }
 41     }
 42 }
 43 
 44 void init(int n)    //一堆初始化。
 45 {
 46     dfn.clear();
 47     for(int i=1; i<=n; i++) vect[i].clear(),pre[i]=i;
 48     for(int i=1; i<=n; i++)
 49         for(int j=1; j<=n; j++)
 50             maxcost[i][j]=0.0;
 51     memset(used, 0, sizeof(used));
 52     memset(vis,  0, sizeof(vis));
 53 }
 54 double cal(int n, int m)
 55 {
 56     init(n);
 57     double sum=0.0;   //MST
 58     for(int i=0; i<m; i++)      //kruscal求最小生成树
 59     {
 60         int u=find(a[seq[i]]);
 61         int v=find(b[seq[i]]);
 62         if( u!=v )
 63         {
 64             pre[u]=v;               //不是同个连通块，则连接。
 65             vect[a[seq[i]]].push_back( b[seq[i]] ); //顺便建图，方便建树
 66             vect[b[seq[i]]].push_back( a[seq[i]] );
 67             used[seq[i]]=1;
 68             sum+=w[seq[i]];
 69         }
 70     }
 71 
 72     DFS(1);                   //求任意点对间的最小瓶颈路
 73     double ans=0.0;
 74     for(int i=0; i<m; i++)    //穷举徐福声将要建的边。
 75     {
 76         double A=p[a[i]]+p[b[i]], B;
 77 
 78         if(used[i])     B=sum-w[i];            //树上的边
 79         else            B=sum-maxcost[a[i]][b[i]];
 80         ans=max( A/B, ans );
 81     }
 82     return ans;
 83 }
 84 
 85 
 86 int main()
 87 {
 88     freopen("input.txt", "r", stdin);
 89     int t, n;
 90     cin>>t;
 91     while(t--)
 92     {
 93         cin>>n;
 94         for(int i=1; i<=n; i++)    scanf("%d%d%d",&x[i],&y[i],&p[i]);
 95         int cnt=0;
 96         for(int i=1; i<=n; i++)     //求两点间的距离,共n*(n-1)/2条边
 97         {
 98             for(int j=i+1; j<=n; j++)
 99             {
100                 a[cnt]=i;
101                 b[cnt]=j;
102                 w[cnt]=dis(i,j);
103                 seq[cnt]=cnt;   //千万不要用seq[cnt]=cnt++;或者seq[cnt++]=cnt。
104                 cnt++;
105             }
106         }
107         sort(seq, seq+cnt, cmp);        //按边长排序
108         printf("%.2f\n", cal(n, cnt));
109     }
110 
111     return 0;
112 }