CSUOJ 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

Hint

简单题，使用set，把第一个人的好友ID存到set里，判断set中是否存在第二人的好友就行了。

#include<stdio.h>

#include<string.h>

#include<queue>

#include<algorithm>

#include<set>

using namespace std;

set<long long>s;

int main()

{

	int T,n,m;

	long long cnt;

	while(~scanf("%d",&T))

	{

		while(T--)

		{

			s.clear();

			int flag=0;

			scanf("%d%d",&n,&m);

			for(int i=0;i<n;i++)

			{

				scanf("%lld",&cnt);

				s.insert(cnt);

			}

			for(int i=0;i<m;i++)

			{

				scanf("%lld",&cnt);

				//printf("%d ",s.count(cnt));

				if(s.count(cnt))

					flag=1;

			}

			if(flag)

				printf("Yes\n");

			else

				printf("No\n");

		}

	}

	return 0;

}

/**********************************************************************

	Problem: 1868

	User: leo6033

	Language: C++

	Result: AC

	Time:688 ms

	Memory:1708 kb

**********************************************************************/