LA 5713 Qin Shi Huang's National Road System 最小生成树

LA 5713

题意：给n个城市的位置以及人口p，所有城市可联通，要求连接两个城市的总人口最大A且其他城市连接的总长度最小B，求最大的A/B。

思路：首先求出图的最小生成树权值sum，然后在最小生成树中用dfs求出任意两点的最小瓶颈路maxcost值，接下来就是遍历所有城市求一个最大的pi+pj/sum-maxcost[ i ][ j ]。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<cmath>
using namespace std;
const int maxn=1e3+5;
struct point
{
	int x,y,p;
}c[maxn];
struct node
{
	int u,v;
	double len;
}a[maxn*maxn];
int n,par[maxn];
vector<int>G[maxn],V;
double dis[maxn][maxn],maxcost[maxn][maxn],sum;
bool cmp(node x,node y)
{
	return x.len<y.len;
}
void init()
{
	for(int i=0;i<=n;i++)
	{
		par[i]=i;
		G[i].clear();
	}
	V.clear();
	sum=0;
}
int find(int x)
{
	if(par[x]!=x)
	par[x]=find(par[x]);
	return par[x];
}
void kruskal()
{
	int i,j,tot=0;
	for(i=1;i<=n*(n-1)/2;i++)
	{
		int u=find(a[i].u);
		int v=find(a[i].v);
		if(u!=v)
		{
			par[u]=v;
			tot++;
			G[a[i].u].push_back(a[i].v);
			G[a[i].v].push_back(a[i].u);
			sum+=a[i].len;
		}
		if(tot==n-1)
		break;
	}
}
void dfs(int u,int fa)
{
	for(int i=0;i<V.size();i++)
	{
		int v=V[i];
		if(v==fa)
		maxcost[v][u]=maxcost[u][v]=dis[u][fa];
		else
		maxcost[v][u]=maxcost[u][v]=max(maxcost[fa][v],dis[u][fa]);
	}
	V.push_back(u);
	for(int i=0;i<G[u].size();i++)
	{
		int v=G[u][i];
		if(v!=fa)
		dfs(v,u);
	}
}
void work()
{
	int i,j;
	double ans=0;
	for(i=1;i<=n;i++)
	for(j=1;j<i;j++)
	{
		double t=c[i].p+c[j].p;
		double t1=sum-maxcost[i][j];
		ans=(double)max(ans,t/t1);
	}
	printf("%.2lf\n",ans);
}
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		int i,j,tot=0;
		scanf("%d",&n);
		init();
		for(i=1;i<=n;i++)
		{
			scanf("%d%d%d",&c[i].x,&c[i].y,&c[i].p);
			for(j=1;j<i;j++)
			{
				a[++tot].u=i;
				a[tot].v=j;
				a[tot].len=sqrt(pow(c[i].x-c[j].x,2)+pow(c[i].y-c[j].y,2));
				dis[i][j]=dis[j][i]=a[tot].len;
			}
		}
		sort(a+1,a+tot+1,cmp);
		kruskal();
		dfs(1,-1);
		work();
	}
}