poj 2349 Arctic Network MST/二分答案

poj 2349 Arctic Network

题目传送

Sol:

方法一：

贪心的想，发现n个点只需要n-1条边即可，求MST即可，再把MST中最大的m-1条边去掉，第m大就是答案。

code：

#include<string>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define IL inline
#define RG register
#define DB double
#define LL long long
using namespace std;

IL int gi() {
    RG int x=0,p=1; RG char ch=getchar();
    while(ch<'0'||ch>'9') {if(ch=='-') p=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+(ch^48),ch=getchar();
    return x*p;
}

const int N=507;

DB d[N];
int n,m,tot,fa[N];

struct DOT{int x,y;}a[N];
struct EDGE{int x,y;DB d;}e[N*N<<1];

IL bool operator <(EDGE A,EDGE B) {return A.d<B.d;}

IL DB dis(int s,int b) {
    return sqrt(1.0*(a[s].x-a[b].x)*(a[s].x-a[b].x)+(a[s].y-a[b].y)*(a[s].y-a[b].y));
}

int getfa(int x) {return fa[x]==x?x:fa[x]=getfa(fa[x]);}

IL void Kruskal() {
    RG int i,fx,fy,cnt=0;
    for(i=1;i<=n;++i) fa[i]=i;
    for(i=1;i<=tot;++i) {
        fx=getfa(e[i].x),fy=getfa(e[i].y);
        if(fx==fy) continue;
        fa[fx]=fy,d[++cnt]=e[i].d;
        if(cnt==n-1) break;
    }
    printf("%.2lf\n",d[n-m]);//poj 上如用G++交代码，则需把%lf改为%f
}

int main()
{
    RG int i,j,T=gi();
    while(T--) {
        m=gi(),n=gi(),tot=0;
        for(i=1;i<=n;++i) a[i].x=gi(),a[i].y=gi();
        for(i=1;i<=n;++i)
            for(j=i+1;j<=n;++j)
                e[++tot]=(EDGE){i,j,dis(i,j)};
        sort(e+1,e+tot+1);
        Kruskal();
    }
    return 0;
}
//poj 145ms

方法二：

考虑二分答案，满足条件的点连好边，然后dfs一下有几个连通块，有多少连通块就需要多少卫星，和m比较即可。

相比方法一，code又慢又长。。。

code：

#include<string>
#include<cmath>
#include<cstdio>
#include<cstring>
#define IL inline
#define RG register
#define DB double
#define LL long long
using namespace std;

IL int gi() {
    RG int x=0,p=1; RG char ch=getchar();
    while(ch<'0'||ch>'9') {if(ch=='-') p=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+(ch^48),ch=getchar();
    return x*p;
}

const int N=505;
const DB eps=1e-4;

DB l,r,mid;
int n,m,cnt,tot,vis[N],head[N];

struct DOT{int x,y;}a[N];
struct EDGE{int next,to;}e[N*N<<1];

IL void New_case() {
    tot=cnt=0;
    memset(&e,0,sizeof(e));
    memset(vis,0,sizeof(vis));
    memset(head,0,sizeof(head));
}

IL void make(int x,int y) {
    e[++tot]=(EDGE){head[x],y},head[x]=tot;
    e[++tot]=(EDGE){head[y],x},head[y]=tot;
}

IL DB dis(int s,int b) {
    return 1.0*sqrt((a[s].x-a[b].x)*(a[s].x-a[b].x)+(a[s].y-a[b].y)*(a[s].y-a[b].y));
}

IL void dfs(int x) {
    RG int i,y;
    vis[x]=1;
    for(i=head[x];i;i=e[i].next)
        if(!vis[y=e[i].to]) dfs(y);
}

IL int check(DB val) {
    RG int i,j;
    New_case();
    for(i=1;i<=n;++i)
        for(j=i+1;j<=n;++j)
            if(dis(i,j)<=val) make(i,j);
    for(i=1;i<=n;++i)
        if(!vis[i]) dfs(i),++cnt;
    return cnt<=m;
}

int main()
{
    RG int i,T=gi();
    while(T--) {
        m=gi(),n=gi();
        for(i=1;i<=n;++i) a[i].x=gi(),a[i].y=gi();
        l=0,r=1e9;
        while(r-l>=eps) {
            mid=(l+r)/2;
            if(check(mid)) r=mid;
            else l=mid;
        }
        printf("%.2lf\n",r);
    }
    return 0;
}
// poj 1400多ms