bzoj2429: [HAOI2006]聰明的猴子(最小生成樹)

Hanks_o發表於2018-03-20

題目傳送門

解法:
最小生成樹嘛。
你要希望越多的猴子能在全圖跑。
那麼點與點之間距離應該儘量小。
那麼就滿足了最小生成樹咯。

所有邊跑出來建最小生成樹。記錄最大邊權即可。。

程式碼實現:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
struct node {int x,y;double c;}a[1110000];int len;
bool cmp(node n1,node n2) {return n1.c<n2.c;}
bool cmp1(int a,int b) {return a>b;}
int fa[1100];int findfa(int x) {if(x!=fa[x])fa[x]=findfa(fa[x]);return fa[x];}
double juli(int x1,int y1,int x2,int y2) {return double(sqrt((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2)));}
int xx[1100],yy[1100],s[1100];
int main() {
    int m;scanf("%d",&m);
    for(int i=1;i<=m;i++)scanf("%d",&s[i]);sort(s+1,s+1+m,cmp1);
    int n;scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d%d",&xx[i],&yy[i]);len=0;
    for(int i=1;i<=n;i++)for(int j=i+1;j<=n;j++) {
        len++;a[len].x=i;a[len].y=j;a[len].c=juli(xx[i],yy[i],xx[j],yy[j]);
    }sort(a+1,a+1+len,cmp);
    for(int i=1;i<=n;i++)fa[i]=i;int t=0;double C;
    for(int i=1;i<=len;i++) {
        int xx=findfa(a[i].x),yy=findfa(a[i].y);
        if(xx!=yy) {
            fa[xx]=yy;
            t++;if(t==n-1) {C=a[i].c;break;}
        }
    }
    for(int i=1;i<=m;i++) if(double(s[i])<C) {printf("%d\n",i-1);return 0;}
    return 0;
}

相關文章