hdu4081次小生成树

先求一遍最小生成树，然后遍历所有边，如果这条边在最小生成树中就直接减去这条边的距离，如果不在最小生成树中，那么就构成了一个环，此时需要减去最小生成树中最大的边，即求次小生成树时的maxx，

有一点要注意当求maxx最大值时j！=k，虽然不知道原理是什么。。。。如果有大佬知道求告知

好像求次小生成树时没有这个条件，据说还能prim+dfs做。。。。

#include<map>

#include<set>

#include<cmath>

#include<queue>

#include<stack>

#include<vector>

#include<cstdio>

#include<cassert>

#include<iomanip>

#include<cstdlib>

#include<cstring>

#include<iostream>

#include<algorithm>

#define pi acos(-1)

#define ll long long

#define mod 1000000007

#define ls l,m,rt<<1

#define rs m+1,r,rt<<1|1

#pragma comment(linker, "/STACK:1024000000,1024000000")

using namespace std;

const double g=10.0,eps=1e-;

const int N=+,maxn=,inf=0x3f3f3f;

struct edge{

    double x,y,p;

}e[N];

double c[N][N],d[N];

double maxx[N][N];

int pre[N],n;

bool vis[N],used[N][N];

double dis(edge a,edge b)

{

    return sqrt((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y));

}

double prim()

{

    memset(vis,,sizeof vis);

    memset(used,,sizeof used);

    memset(maxx,,sizeof maxx);

    for(int i=;i<=n;i++)

    {

        pre[i]=;

        d[i]=c[][i];

    }

    vis[]=;

    pre[]=;

    d[]=;

    double ans=;

    for(int i=;i<n;i++)

    {

        double mind=inf;

        int k;

        for(int j=;j<=n;j++)

        {

            if(!vis[j]&&mind>d[j])

            {

                mind=d[j];

                k=j;

            }

        }

        vis[k]=;

        ans+=mind;

        used[k][pre[k]]=used[pre[k]][k]=;

        for(int j=;j<=n;j++)

        {

            if(vis[j]&&k!=j)maxx[j][k]=maxx[k][j]=max(maxx[j][pre[k]],d[k]);

            if(!vis[j]&&d[j]>c[k][j])

            {

                d[j]=c[k][j];

                pre[j]=k;

            }

        }

    }

    return ans;

}

int main()

{

    ios::sync_with_stdio(false);

    cin.tie();

    cout<<setiosflags(ios::fixed)<<setprecision();

    int t;

    cin>>t;

    while(t--){

        cin>>n;

        for(int i=;i<=n;i++)

            cin>>e[i].x>>e[i].y>>e[i].p;

        for(int i=;i<=n;i++)

        {

            c[i][i]=;

            for(int j=i+;j<=n;j++)

            {

                c[i][j]=c[j][i]=dis(e[i],e[j]);

            }

        }

        double B=prim(),ans=-;

        for(int i=;i<=n;i++)

        {

            for(int j=i+;j<=n;j++)

            {

                if(used[i][j])ans=max(ans,(e[i].p+e[j].p)/(B-c[i][j]));

                else ans=max(ans,(e[i].p+e[j].p)/(B-maxx[i][j]));

            }

        }

        cout<<ans<<endl;

    }

    return ;

}

巴特西

hdu4081次小生成树

最新文章

热门文章