小米2013年校园招聘笔试题——朋友圈

最新推荐文章于 2024-05-31 21:15:25 发布

weixin_30352645

最新推荐文章于 2024-05-31 21:15:25 发布

阅读量391

点赞数

文章标签：测试 runtime

原文链接：http://www.cnblogs.com/unclejelly/p/4082155.html

版权

题目描述：

假如已知有n个人和m对好友关系（存于数字r）。如果两个人是直接或间接的好友（好友的好友的好友...），则认为他们属于同一个朋友圈，请写程序求出这n个人里一共有多少个朋友圈。
假如：n = 5 ， m = 3 ， r = {{1 , 2} , {2 , 3} , {4 , 5}}，表示有5个人，1和2是好友，2和3是好友，4和5是好友，则1、2、3属于一个朋友圈，4、5属于另一个朋友圈，结果为2个朋友圈。

输入：

输入包含多个测试用例，每个测试用例的第一行包含两个正整数 n、m，1=<n，m<=100000。接下来有m行，每行分别输入两个人的编号f，t（1=<f，t<=n），表示f和t是好友。当n为0时，输入结束，该用例不被处理。

输出：

对应每个测试用例，输出在这n个人里一共有多少个朋友圈。

样例输入：

样例输出：

2
1

来源：

小米2013年校园招聘笔试题

这是一道基础的并查集问题。要注意的是数组要开大点，否则跟我一样Runtime Error就悲剧了。

#include <iostream>
using namespace std;
const int MAX=300000;
int n,m,k;
int parent[MAX+10];
int bit[MAX+10];
int GetParent(int a)
{
	if(parent[a]!=a)
		parent[a]=GetParent(parent[a]);
	return parent[a];
}
void Merge(int a,int b)
{
	int p1=GetParent(a);
	int p2=GetParent(b);
	if(p1==p2)
		return;
	bit[p2]=0;
	parent[p2]=p1;
}
int main()
{
	int i;
	while(1)
	{
		int f,t;
		cin>>n;
		if(n==0)
			break;
		cin>>m;
		bit[0]=0;
		for(i=1;i<=n;i++)
			{
				parent[i]=i;
				bit[i]=1;
	    	}
		for(i=0;i<m;i++)
		{
			cin>>f>>t;
			Merge(f,t);
		}
		int sum=0;
		for(i=1;i<=n;i++)
		{
			if(bit[i])
				sum++;
		}
		cout<<sum<<endl;
	}
	return 0;
}

转载于:https://www.cnblogs.com/unclejelly/p/4082155.html