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
#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
**********************************************************************/