poj1751:
题意:已知某些边建好的情况下求最小生成树中其他需要建的边,输出其起点和终点
这道题目是spj,对输出的顺序没有要求
我用java写一直MLE,百思不得其解,然后又用C++写了一遍就AC了。
这次终于知道了在结构体内写比较函数要比在外面写cmp快的多
第二次在外面写的cmp直接卡时间超低空飘过,呵呵
不多说了,看代码吧
//============================================================================
// Name : poj1751Highways.cpp
// Author : xinge
// Version :
// Copyright : Your copyright notice
// Description : nsi-style
//============================================================================
#include <iostream>
#include <stdio.h>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <math.h>
using namespace std;
const int Max=800;
int N,M;
int parent[Max];
struct point
{
int x,y;
}data1[Max];
struct edge
{
int start,end;
double val;
bool operator <(const edge a)const
{
return val<a.val;
}
}data2[(Max*Max)];
//就是这个卡时间
//bool cmp(edge e1,edge e2)
//{
// return e1.val<e2.val;
//}
void init()
{
for(int i=0;i<Max;i++)
parent[i]=i;
}
int find(int x)
{
if(parent[x]!=x)
parent[x]=find(parent[x]);
return parent[x];
}
void merge(int a,int b)
{
int fa=find(a);
int fb=find(b);
parent[fa]=fb;
}
void kruskal(int len)
{
int i,fa,fb;
for(i=1;i<=len;i++)
{
fa=find(data2[i].start);
fb=find(data2[i].end);
if(fa!=fb)
{
printf("%d %d\n",data2[i].start,data2[i].end);
merge(data2[i].start,data2[i].end);
}
}
}
int main() {
scanf("%d",&N);
int i;
for(i=1;i<=N;i++)
scanf("%d%d",&(data1[i]).x,&(data1[i]).y);
init();
scanf("%d",&M);
int j,a,b;
for(j=1;j<=M;j++)
{
scanf("%d%d",&a,&b);
merge(a,b);
}
int len=0;
for(i=1;i<=N;i++)
for(j=i+1;j<=N;j++)
{
len++;
data2[len].start=i;
data2[len].end=j;
data2[len].val=sqrt((double)((data1[i].x-data1[j].x)*(data1[i].x-data1[j].x)+(data1[i].y-data1[j].y)*(data1[i].y-data1[j].y)));
}
sort(data2+1,data2+len+1);
kruskal(len);
return 0;
}