您的位置:首页 > 其它

POJ 1861||ZOJ1542 kruskal求最小生成树(POJ秒杀啊)

2013-03-25 14:49 459 查看
题意:用所给的网线链接所有的路流器,使得最长的单根网线的长度在所有方案中是最小的。

分析:本质还是求最小生成树。为了输出单根网线的长度在所有方案中是最小的以及输入p对顶点,需要在选择的时候记录下已选则 的网线的下标。

zoj这题80ms,而uva 前面几次tle。后面人品好点干好1s。啊。秒杀啊。擦。

View Code

// I'm the Topcoder
//C
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include <math.h>
#include <time.h>
//C++
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <cctype>
#include <stack>
#include <string>
#include <list>
#include <queue>
#include <map>
#include <vector>
#include <deque>
#include <set>
using namespace std;

//*************************OUTPUT*************************
#ifdef WIN32
#define INT64 "%I64d"
#define UINT64 "%I64u"
#else
#define INT64 "%lld"
#define UINT64 "%llu"
#endif

//**************************CONSTANT***********************
#define INF 0x3f3f3f3f
#define eps 1e-8
#define PI acos(-1.)
#define PI2 asin (1.);
typedef long long LL;
//typedef __int64 LL;   //codeforces
typedef unsigned int ui;
typedef unsigned long long ui64;
#define MP make_pair
typedef vector<int> VI;
typedef pair<int, int> PII;
#define pb push_back
#define mp make_pair

//***************************SENTENCE************************
#define CL(a,b) memset (a, b, sizeof (a))
#define sqr(a,b) sqrt ((double)(a)*(a) + (double)(b)*(b))
#define sqr3(a,b,c) sqrt((double)(a)*(a) + (double)(b)*(b) + (double)(c)*(c))

//****************************FUNCTION************************
template <typename T> double DIS(T va, T vb) { return sqr(va.x - vb.x, va.y - vb.y); }
template <class T> inline T INTEGER_LEN(T v) { int len = 1; while (v /= 10) ++len; return len; }
template <typename T> inline T square(T va, T vb) { return va * va + vb * vb; }

// aply for the memory of the stack
//#pragma comment (linker, "/STACK:1024000000,1024000000")
//end

int n,m;
#define maxn 1000+10
#define maxm 15000+100
int pa[maxn];
struct node{
int u,v,w;
}edges[maxm];

int sum=0;
int ans[maxn],k;
int maxedge=-INF;
//初始化
void UFset(){
for(int i=1;i<=n;i++){
pa[i]=-1;
}
}

//查找
int findset(int x){
int s;
for(s=x;pa[s]>=0;s=pa[s]);
while(s!=x){
int tmp=pa[x];
pa[x]=s;
x=tmp;
}
return s;
}

//合并
void Union(int R1,int R2){
int r1=findset(R1), r2=findset(R2);
//int count=0;
int tmp=pa[r1]+pa[r2];
if(pa[r1]>pa[r2]){
pa[r1]=r2;
pa[r2]=tmp;
}
else {
pa[r2]=r1;
pa[r1]=tmp;
}
}

void Kruskral(){
//初始化
UFset();
sum=0;
k=0;
int u,v;
for(int i=0;i<m;i++){
u=edges[i].u; v=edges[i].v;
if(findset(u)!=findset(v)){
ans[k]=i;k++;
if(edges[i].w>maxedge)  maxedge=edges[i].w;
sum++;
Union(u,v);

}
if(sum>=n-1) break;
}
}

int cmp(const void*a,const void*b){
node aa=*(const node*)a;
node bb=*(const node*)b;
if(aa.w>bb.w)  return 1;
else return -1;
}

int main(){
while(scanf("%d%d",&n,&m)!=EOF){
memset(ans,0,sizeof(ans));
for(int i=0;i<m;i++){
scanf("%d%d%d",&edges[i].u,&edges[i].v,&edges[i].w);
}
qsort(edges,m,sizeof(edges[0]),cmp);
sum=0;maxedge=-INF;
Kruskral();
printf("%d\n",maxedge);
printf("%d\n",sum);
for(int i=0;i<sum;i++){
printf("%d %d\n",edges[ans[i]].u,edges[ans[i]].v);
}

}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: