1868: 潜在好友

1868: 潜在好友

Description

小X在搬砖写一个论坛，这个时候老板突然想到一个功能，让小X今天赶快实现。大概就是如果某个人是你好友的好友那么他的头像上面会有特殊的标志。小X想不到较好的办法来解决如何验证两个人是不是好友的好友，现在向你求助。

Input

第一行是一个整数T(1<=T<=100)，代表数据组数。每组数据第一行是两个整数n,m(1<=n,m<=10000)，代表这两个人的好友的数量。之后n行是第一个人的好友id再之后m行是第二个人的好友id(1<=id<=1e9)

Output

如果第二个人是第一个人的好友的好友输出Yes否则No

Sample Input

2
1 1
1
2
2 3
1
2
2
3
4

Sample Output

No
Yes

这个题目说白了就是找是否存在相同的元素。看到这题我的第一想法就是用set存储所有的好友，然后判断set里的元素个数是否等于两个人的好友数之和，因为set集合不会有相同元素出现。但很不幸的错了，而且我一直没找到错在哪。下面是错误的代码，如果哪位大佬知道为什么，还麻烦评论区说一声，感谢。

#include<iostream>
#include<set>
using namespace std;
set<int> s;
int main()
{
	int T;
	cin >> T;
	while (T--)
	{
		s.clear();
		int m, n;
		scanf ("%d %d", &m, &n);
		int x;
		int flag = 0;
		for (int i = 0; i < m + n; i++)
		{
			scanf ("%d", &x);
			s.insert(x);
		}
		if (s.size() < m + n) printf("Yes\n");
		else printf("No\n");
	}
	return 0;
}
/**********************************************************************
	Problem: 1868
	User: pengpeng
	Language: C++
	Result: WA
**********************************************************************/

下面的代码是正确的了。思路是在第二个人的好友里找是否有已经出现过的，即共同好友，如果有就标记为1，最后判断标记是否为真就好了。顺便说一句，需要用scanf，如果用cin会TLE。

#include<iostream>
#include<set>
using namespace std;
set<int> s;
int main()
{
	int T;
	cin >> T;
	while (T--)
	{
		s.clear();
		int m, n;
		scanf ("%d %d", &m, &n);
		//cin >> m >> n;
		int x;
		int flag = 0;
		for (int i = 0; i < m; i++)
		{
			scanf ("%d", &x);
			//cin >> x;
			s.insert(x);
		}
		for (int i = 0; i < n; i++)
		{
			scanf ("%d", &x);
			//cin >> x;
			if (s.count(x))flag = 1;
		}
		if (flag) printf("Yes\n");
		else printf("No\n");
		//if (flag)cout << "Yes\n";
		//else cout << "No\n";
	}
	return 0;
}
/**********************************************************************
	Problem: 1868
	User: pengpeng
	Language: C++
	Result: AC
	Time:696 ms
	Memory:2552 kb
**********************************************************************/