ZOJ-2833

并查集,直接套的模板,不过内存好大。。好像空间优化不好,先这样吧

#include<stdio.h>
#include<stdlib.h>
#include<string.h>

struct FriendStruct
{
	int id;
	struct FriendStruct *parent;
	int rank;
};

typedef struct FriendStruct *Friend;

static Friend make_set(int id)
{
	Friend stu = malloc(sizeof(struct FriendStruct));
	stu->parent = stu;
	stu->id = id;
	stu->rank = 1;
	return stu;
}

static Friend find_set(Friend stu)
{
	if (stu->parent != stu)
		stu->parent = find_set(stu->parent);
	return stu->parent;
}

static void link(Friend s1, Friend s2)
{
	if (s1->rank > s2->rank)
	{
		s2->parent = s1;
		s1->rank += s2->rank;
	}
	else
	{
		s1->parent = s2;
		s2->rank += s1->rank;
	}
}

static void union_set(Friend s1, Friend s2)
{
	link(find_set(s1), find_set(s2));
}

int main()
{
	int n, m, count = 0;
	Friend *array = malloc(100000 * sizeof(Friend));
	while (scanf("%d %d", &n, &m) != EOF)
	{
		memset(array, 0, 100000 * sizeof(Friend));
		getchar();
		if (count++)
			putchar('\n');
		printf("Case %d:\n", count);

		int i, a, b;
		char mq;
		for (i = 0; i < m; i++)
		{
			mq = getchar();
			if (mq == 'M')
			{
				scanf("%d %d", &a, &b);
				getchar();
				if (array[a - 1] == NULL)
					array[a - 1] = make_set(a);
				if (array[b - 1] == NULL)
					array[b - 1] = make_set(b);
				if (find_set(array[a - 1]) != find_set(array[b - 1]))
					union_set(array[a - 1], array[b - 1]);
			}
			else
			{
				scanf("%d", &a);
				getchar();
				if (array[a - 1] == NULL)
					puts("1");
				else
					printf("%d\n", find_set(array[a - 1])->rank);
			}
		}

	}
	free(array);
	return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值