hdu 4081 Qin Shi Huang's National Road System (prim+lca)

hdu 4081

描述： 给定n个城市坐标和人口（X，Y，P）。求max（A/B）： A=Pi+Pj, B=Wi（i所在的生成树权值和）+Wj（j所在的生成树权值和）（i，j在两颗不同的生成树，两颗生成树包含所有n个点）

思路：枚举i，j，求关于i，j的次小生成树即可，因为次小生成树的权值和一定是最小的

先通过prim求出最小生成树，再使用lca

代码如下：187ms

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <queue>
#include <vector>
#include <stack>
#include <iostream>
#include <map>
#include <string>
#include <cmath>
using namespace std;
typedef long long ll;
const int maxn = 1e3+1;

int N;
double x[maxn], y[maxn], p[maxn];
double dis(int a, int b)
{
    return sqrt((x[a]-x[b])*(x[a]-x[b])+(y[a]-y[b])*(y[a]-y[b]));
}
double mst;
int vis[maxn];
struct node
{
	int to;double w;
	node() {}
	node(int to, double w):to(to), w(w) {}
};
vector<node>vec[maxn];
void add(int a, int b, double w)
{
    vec[a].push_back(node(b,w));
    vec[b].push_back(node(a,w));
}

void prim()
{
	mst=0;
	double d[maxn];
	int pre[maxn];
	for(int i=2;i<=N;i++) d[i]=dis(i, 1), pre[i]=1;
	memset(vis, 0, sizeof vis); vis[1]=1;
	for(int kk=1;kk<N;kk++)
	{
		int k=0; double mi=1e8;
		for(int i=2;i<=N;i++) if(!vis[i] && d[i]<mi)
		{
			mi=d[i]; k=i;
		}
		vis[k]=1;
		mst+=mi;

		add(k, pre[k], mi);

		for(int i=2;i<=N;i++) if(!vis[i] && d[i]>dis(i, k))
		{
			d[i]=dis(i, k);
			pre[i]=k;
		}
	}
}
int f[maxn];
double m[maxn];
int find(int x)
{
	if(f[x]==x) return x;
	int pf=f[x];
	f[x]=find(f[x]);
	m[x]=max(m[x], m[pf]);
	return f[x];
}
double mat[maxn][maxn];
int fa[maxn][maxn];
vector<int>ans[maxn];
void lca(int u)
{
	vis[u]=1;

	for(int i=0;i<vec[u].size();i++)
	{
		int j=vec[u][i].to;
		if(!vis[j])
		{
			lca(j);
			f[j]=u;
			m[j]=vec[u][i].w;
		}
	}

	for(int l=1;l<=N;l++)
		if(!fa[u][l]) fa[u][l]=fa[l][u]=1;
		else
		{
			int ff=find(l);
			ans[ff].push_back(l);
			ans[ff].push_back(u);
		}

	for(int i=0;i<ans[u].size();i+=2)
	{
		int a=ans[u][i];
		int b=ans[u][i+1];
		mat[b][a]=mat[a][b]=max(m[a], m[b]);
	}
}

int main()
{
	int T; cin>>T;
	while(T--)
	{
		cin>>N;
		for(int i=1;i<=N;i++) scanf("%lf%lf%lf", x+i, y+i, p+i);
		for(int i=1;i<=N;i++) vec[i].clear();
		prim();

		memset(vis, 0, sizeof vis);
		memset(fa, 0, sizeof fa);
		for(int i=1;i<=N;i++) f[i]=i;
	//	memset(mat, 0, sizeof mat);
        for(int i=1;i<=N;i++) ans[i].clear();
		memset(m, 0, sizeof m);
		lca(1);
/*
        for(int i=1;i<=N;i++)
        {
            for(int j=1;j<=N;j++)
            printf("%.2lf ", mat[i][j]);
            putchar('\n');
        }

       for(int i=1;i<=N;i++) printf("%d ", p[i]); putchar('\n');
*/
        double ANS=0;

    //    printf("%lf\n", mst);

        for(int i=1;i<=N;i++) for(int j=1;j<i;j++)
            ANS=max(ANS, (p[i]+p[j])/(mst-mat[i][j]));

        printf("%.2lf\n", ANS);

	}
    return 0;
}