累觉不爱POJ 1751Prim

原创 2014年09月19日 13:18:51

据说这是一道很裸的Prim模板题,网上的解释也很多。倔强的我在看了Prim生成原理后也决定自己动手写了。

STEP 1

已经相连的城市放在一个集合sele里,没选的放在队列remain里,然后每次都在remain里取出一个点,看看这个点与sele里的点哪个距离最小。输出。再把这个点归入sele。

#include <cstdio>
#include <iostream>
#include <cmath>
#include <cstring>
#include <queue>
#include <algorithm>
using namespace std;
#define N 751
#define M 1001
int n, m;
struct node
{
	int x, y;
	double dis[N];
}city[N];
bool sele[N];
queue <int>remain;
int ans;
int main()
{
	int i, j, count;
	double min;
	while(~scanf("%d", &n))
	{
		count = 0;
		for(i = 1;i <= n;i++)
			scanf("%d%d", &city[i].x, &city[i].y);
		for(i = 1;i <= n;i++)
			for(j = 1;j <= n;j++)
				city[i].dis[j] = sqrt(pow(city[i].x-city[j].x,2)+pow(city[i].y-city[j].y,2));
		scanf("%d", &m);
		memset(city, 0, n);
		while(m--)
		{
			scanf("%d%d", &i, &j);
			sele[i] = 1;
			sele[j] = 1;
		}
		for(i = 1;i <= n;i++)
		{
			if(!sele[i])
				remain.push(i);
		}
		while(!remain.empty())
		{
			min = 1e5;
			j = remain.front();
			for(i = 1;i <= n;i++)
			{
				if(sele[i])
				{
					double t = city[i].dis[j];
					if(t < min)
					{
						min = t;
						ans = i;
					}
				}
			}
			sele[j] = 1;
			printf("%d %d\n", ans, j);
			remain.pop();
		}
	}
	return 0;
}
万万没想到,这不是prim啊!prim取的是在sele里的所有点向外连的所有线段里的最小线段。而我却是在remain里选点,显然不是滴……

而且这样的算法还会漏掉几条边。比如说1-3,2-5连起来了,1,2,3,5都被放在了sele里。如果刚好1-4连好,一切OK,那么1-3-4在一起,2-5在一起,形成了两个集合,但是它们之间却不连通呢!


STEP 2

愚蠢的我为了解决不连通的缺陷,特别设置了数组链表vector vet[n],把同在一个集合里的城市放在一个链表里,并且用order记录每个城市所在的vet序号,只有vet[1]被当作sele,当有其他vet里的城市被连进vet[1]时,该vet里的城市都被连vet[1]。由于找min是O(N^3)时间复杂度,TLE了。

进步在于这用了prim思想,确实找到了vet[1]里的所有点向外连的所有线段里的最小线段。

#include <cstdio>
#include <iostream>
#include <cmath>
#include <cstring>
#include <queue>
#include <algorithm>
#include <vector>
using namespace std;
#define N 751
#define M 1001
int n, m;
struct node
{
	int x, y;
	double dis[N];
}city[N];
int order[N];
int main()
{
	vector<int>vet[N];
	int i, j, k, point;
	double min;
	while(~scanf("%d", &n))
	{
		for(i = 1;i <= n;i++)
			scanf("%d%d", &city[i].x, &city[i].y);
		for(i = 1;i <= n;i++)
			for(j = 1;j <= n;j++)
				city[i].dis[j] = sqrt(pow(city[i].x-city[j].x,2)+pow(city[i].y-city[j].y,2));
		scanf("%d", &m);
		for(i = 1;i <= n;i++)
			order[i] = 0;
		for(i = 1;i <= point;i++)
			while(!vet[i].empty())
				vet[i].pop_back();
		point = 1;
		int temp, big, change, len;
		if(m)
		{
			scanf("%d%d", &i, &j);
			order[i] = 1;
			order[j] = 1;
			vet[1].push_back(i);
			vet[1].push_back(j);
			m--;
			while(m--)
			{
				scanf("%d%d", &i, &j);
				if(!order[i] && !order[j])
				{
					point++;
					order[i] = point, order[j] = point;
					vet[point].push_back(i);
					vet[point].push_back(j);
				}
				else if(order[i] && !order[j])
				{
					order[j] = order[i];
					vet[order[i]].push_back(j);
				}
				else if(!order[i] && order[j])
				{
					order[i] = order[j];
					vet[order[j]].push_back(i);
				}
				else
				{
					temp = order[i]<order[j]?order[i]:order[j];
					big = max(order[i], order[j]);
					len = vet[big].size();
					for(k = len - 1;k >= 0;k--)
					{
						change = vet[big][k];
						order[change] = temp;
						vet[big].pop_back();
						vet[temp].push_back(change);
					}
				}
			}
		}
		else
		{
			order[1] = 1;
			vet[1].push_back(1);
		}
		int count = vet[1].size();
		int head, tail, num;
		while(count < n)
		{
			double min = 1e5;
			for(i = 0;i < count;i++)
			{
				num = vet[1][i];
				for(j = 1;j <= n;j++)
				{
					if(order[j] != 1)
					{
						if(city[num].dis[j] < min)
						{
							min = city[num].dis[j];
							head = num, tail = j;
						}
					}
				}
			}
			if(!order[tail])
			{
				order[tail] = 1;
				vet[1].push_back(tail);
			}
			else
			{
				num = order[tail];
				len = vet[num].size();
				for(k = len - 1;k >= 0;k--)
				{
					change = vet[num][k];
					vet[1].push_back(change);
					order[change] = 1;
					vet[num].pop_back();
				}
			}
			printf("%d %d\n", head, tail);
			count = vet[1].size();
		}
	}
	return 0;
}

STEP 3

其实只要把可能要连入的边排序完了一遍再依次找就行了,所以又改进:

#include <cstdio>
#include <iostream>
#include <cmath>
#include <cstring>
#include <queue>
#include <algorithm>
#include <vector>
using namespace std;
#define N 751
#define M 1001
struct node
{
	int x, y;
	long long dis[N];
}city[N];
struct nod
{
	int x, y;
	long long dis;
}ans[N*N];
bool cmp(nod a, nod b)
{
	return a.dis < b.dis;
}
vector<int>vet[N];
int n, m, i, j, k, point, order[N];
int input()
{
	for(i = 1;i <= n;i++)
		scanf("%d%d", &city[i].x, &city[i].y);
	for(i = 1;i <= n;i++)
		for(j = 1;j <= n;j++)
			city[i].dis[j] = pow(city[i].x-city[j].x,2)+pow(city[i].y-city[j].y,2);
	scanf("%d", &m);
	for(i = 1;i <= n;i++)
		order[i] = 0;
	for(i = 1;i <= point;i++)
		while(!vet[i].empty())
			vet[i].pop_back();
}
int bute()
{
	int temp, big, change, len;
	if(!order[i] && !order[j])
	{
		point++;
		order[i] = point, order[j] = point;
		vet[point].push_back(i);
		vet[point].push_back(j);
	}
	else if(order[i] && !order[j])
	{
		order[j] = order[i];
		vet[order[i]].push_back(j);	
	}
	else if(!order[i] && order[j])
	{
		order[i] = order[j];
		vet[order[j]].push_back(i);
	}
	else
	{
		temp = order[i]<order[j]?order[i]:order[j];
		big = max(order[i], order[j]);
		len = vet[big].size();
		for(k = len - 1;k >= 0;k--)
		{
			change = vet[big][k];
			order[change] = temp;
			vet[big].pop_back();
			vet[temp].push_back(change);
		}
	}
}
int merge()
{
	scanf("%d%d", &i, &j);
	order[i] = 1;
	order[j] = 1;
	vet[1].push_back(i);
	vet[1].push_back(j);
	m--;
	while(m--)
	{
		scanf("%d%d", &i, &j);
		bute();
	}
}
int setsum()
{
	m = 0;
	for(i = 1;i <= n;i++)
		for(j = i;j <= n;j++)
			if(i != j && !(order[i] == 1 && order[j] == 1))
			{
				ans[m].x = i;
				ans[m].y = j;
				ans[m].dis = city[i].dis[j];
				m++;
			}
	sort(ans, ans + m, cmp);
}
int main()
{
	while(~scanf("%d", &n))
	{
		int v = 0;
		input();
		point = 1;
		if(!m)
		{
			order[1] = 1;
			vet[1].push_back(1);
		}	
		else
			merge();
		setsum();
		for(k = 0;k < m;k++)
		{
			i = ans[k].x, j = ans[k].y;
			if(order[i] == order[j])
				continue;
			bute();
			printf("%d %d\n", i, j, v = 1);
		}
		if(!v)
			printf("\n");
	}
	return 0;
}
竟然WA了....


STEP 4

经师兄指点,我的Prim理解错误,所以还是先看书吧( •̀ ω •́ )y ~~等我AC归来~~


万没想到,这不是prim啊
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Baileys0530/article/details/39396845

累觉不爱

人人都带着虚伪的面具,人性,taire
  • u014639822
  • u014639822
  • 2014-04-10 16:06:14
  • 107

觉累不爱

写博客时候只要点击切换到MarkDown编辑器,浏览器直接卡死,电脑风扇狂转。心好塞,以为是电脑QQ浏览器兼容性问题,换了谷歌,火狐问题依旧。网搜了下,上一年遇到这问题的不少,现在依旧未解决,效率不敢...
  • u012171129
  • u012171129
  • 2017-03-29 15:11:00
  • 48

2014 项目经理-累觉不爱

不知
  • super_marioli
  • super_marioli
  • 2014-04-28 16:46:31
  • 493

微信开发一年,累觉不爱

我没想到,其实,我又有想到的,公司招我入来负责微信开发,到最后,虽说不上功成身退,但我也没有留下一个烂摊子、一个残局, 可能程序员做到这种地步,也算是一种悲哀了。        深有感触,深有...
  • jilongliang
  • jilongliang
  • 2014-04-23 11:20:10
  • 939

高情商就是说话让人感到舒服

低情商者们充满善意去做一件“好事”,却还让别人很不舒服、很不领情,甚至是对他极度反感。低情商者自己说话做事不识趣,缺乏“通感”,不会换位思考。 高情商就是说话让人感到舒服   你身边有没有...
  • frankchen99
  • frankchen99
  • 2015-09-29 10:32:51
  • 548

不爱,就请收起你的暖昧

本文出自网易社区(club.163.com),如需转载,请联系原作者或网易 ...
  • nearblog
  • nearblog
  • 2006-06-25 20:51:00
  • 645

【湿不了就是不爱了吗?做爱时的阴…

【湿不了就是不爱了吗?做爱时的阴道秘密】有一个问题藏在我心中很久了,那就是随著年纪渐长之后,我觉得阴道越来越乾涩,成为名副其实的「乾妹妹」。因为阴道太乾了,完全没有润滑的感觉,每次跟先生爱爱时都会觉得...
  • maxdbs
  • maxdbs
  • 2017-08-30 15:32:36
  • 281

男人为什么不爱逛街

  因为男人是理智的。理智的人办事总得符合逻辑。逛街自然是为了进商场,进商场自然是为了购物,购物自然是家里使用,使用自然是有需求,反之,家里没有需求,何必购物、何必进商场、何必去逛街?没有需求的情况下...
  • tom0421
  • tom0421
  • 2006-04-10 13:24:00
  • 873

一个不爱看书的人记录一下思念看的书

经常看到文章某某人一年都多少多少本书,以及读书的好处,每每看到这些我都会想想自己完完整整看完的书,两只手就可以数过来。前段时间图书馆借书统计第一名的同学一年借了500多本书,深感震撼,虽然他500多本...
  • kh896424665
  • kh896424665
  • 2017-06-13 22:49:34
  • 310

念起即断,念起不随,念起即觉,觉之既无,

这是禅定时的法门.所谓“念”,指的是我们内心因境、因情等各种不同的状况而浮现出来的念头,这些念头既是入门时的大障碍,也是修成以后的资粮.其次,你要了解什么叫做“觉”,这里特指“觉知”,但是觉知很难解释...
  • happyrabbit456
  • happyrabbit456
  • 2016-10-26 15:13:49
  • 1020
收藏助手
不良信息举报
您举报文章:累觉不爱POJ 1751Prim
举报原因:
原因补充:

(最多只允许输入30个字)