码迷,mamicode.com
首页 > 其他好文 > 详细

HDU 3635 Dragon Balls 七龙珠 Union Find算法

时间:2014-07-28 16:34:33      阅读:350      评论:0      收藏:0      [点我收藏+]

标签:blog   io   2014   for   ar   算法   amp   res   

孙悟空要寻找七龙珠,这回是七龙珠的增强版了,因为这些龙珠会衍生,最后不止七颗龙珠了。

悟空带着布玛的龙珠雷达探测器出发了,却发现布玛的龙珠雷达探测器的程序太垃圾了,所以找到我们这些ACM高手为龙珠雷达探测器写个程序,要求可以显示某颗龙珠所在的城市的位置,该龙珠所在的城市共有多少颗龙珠,龙珠移动过的次数。

布玛是个有钱人啊,写个程序我要价5百万,不算过分吧。因为本程序需要用到Union Find(并查集)算法,而且最困难的部分是如何压缩路径,不压缩路径自然容易做到,要压缩路径可以使得程序加快很多,故此要价5百万,O(∩_∩)O哈哈~。


路径压缩的思路是:

增加一个额外的移动记录次数,方便查找父母节点的时候更新孩子节点。


压缩路径之后程序块上100ms+;效果还是可以的。

不压缩路径的话,就直接接在一棵树上,计算移动次数的时候,直接计算到父母节点的距离就可以了。

Union Find的灵活运用,优化好的话,达到5星级难度的题目。

#include <stdio.h>

const int MAX_N = 10001;
struct Subset
{
	int xp, yNumBall, zTrans, addTrans;
};

Subset subs[MAX_N];
int N;

void initSubs()
{
	for (int i = 1; i <= N; i++)
	{
		subs[i].xp = 0;
		subs[i].yNumBall = 1;
		subs[i].zTrans = 0;
		subs[i].addTrans = 0;//the main help for path compression
	}
}

int findParent(int x)//with path compression
{
	if (subs[x].xp)	//do not use 0 index.
	{
		int p = subs[x].xp;
		subs[x].xp = findParent(p);	//update parent
		
		subs[x].zTrans += subs[p].addTrans;//careful:Update transportation times
		subs[x].addTrans += subs[p].addTrans;	//Watch out: +=, not =
		return subs[x].xp;
	}
	return x;
}

int findParent_2(int x)//without path compression
{
	int p = x;
	if (subs[x].xp)	//do not use 0 index.
	{
		p = findParent(subs[x].xp);	//update parent
		subs[x].zTrans = subs[subs[x].xp].zTrans+1;//Update transportation times
	}
	return p;
}

void unionTwo(int x, int y)
{
	int xp = findParent(x);
	int yp = findParent(y);
	if (xp == yp) return ;

	subs[xp].xp = yp;	//It should be yp, not y. add to its parent directly
	subs[yp].yNumBall += subs[xp].yNumBall;	//only record balls in parent
	subs[xp].addTrans++;	//transportation happen, add one more trans
	subs[xp].zTrans++;	//Watch out: update self's trans
}

int main()
{
	int T, Q, a, b;	//N cities and balls, Q quests
	char cmd;
	scanf("%d", &T);
	for (int t = 1; t <= T; t++)
	{
		scanf("%d %d", &N, &Q);
		initSubs();

		printf("Case %d:\n", t);
		while (Q--)
		{
			getchar();	// get rid of '\n'
			scanf("%c %d", &cmd, &a);
			if (cmd == 'T')
			{
				scanf("%d", &b);
				unionTwo(a, b);
			}
			else
			{
				int ap = findParent(a);
				printf("%d %d %d\n", ap, subs[ap].yNumBall, subs[a].zTrans);
			}//balls recorded in parent, trans times recorded in self node
		}
	}
	return 0;
}



HDU 3635 Dragon Balls 七龙珠 Union Find算法,布布扣,bubuko.com

HDU 3635 Dragon Balls 七龙珠 Union Find算法

标签:blog   io   2014   for   ar   算法   amp   res   

原文地址:http://blog.csdn.net/kenden23/article/details/38192731

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!