-
A寻找同性朋友C,B寻找同性朋友D,检测C与D是否为朋友
注意C和D都是中间人,即C不能是B本人,而D不能是A本人 -
如果用int读入数据,会出现+0000和-0000的情况,不能判断准确判断性别
因此我们需要使用字符串读入数据,注意字符数组的大小不要吝啬 -
N<=300但是id从0一直到9999(测试点2、3都会涉及到9999,不要问我怎么知道的)
为了避免TLE,我用数组另存了出现的id,这样遍历的范围就可以大幅度缩小 -
输出要保留4位小数%04d对应
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
using namespace std;
const int N = 300;
int n, m, K;
int id[N+1];
int gender[10001];
int hv[10001] = { 0 };
int way[10001][10001] = {0};
int getGender(char* s) {
int ans = 0;
int i = 0;
while (i < strlen(s)) {
if (s[i] != '-')
ans = ans * 10 + s[i] - '0';
i++;
}
if (s[0] == '-') gender[ans] = 2;
else gender[ans] = 1;
return ans;
}
int lst[10001][2];
int find(int A, int B) {
int C, D;
int ans = 0;
for (int i = 1; i <= n; ++i)
if (gender[A]==gender[id[i]] && way[A][id[i]] && id[i]!=B) {
C = id[i];
for (int j = 1; j <= n; ++j)
if (i!=j && gender[B] == gender[id[j]] && way[B][id[j]] && id[j]!=A) {
D = id[j];
if (way[C][D]) {
ans++;
lst[ans][0] = C;
lst[ans][1] = D;
}
}
}
return ans;
}
int main()
{
scanf("%d%d", &n, &m);
for (int i = 0; i < m; ++i) {
char s1[10], s2[10];
scanf("%s %s", &s1, &s2);
int x = getGender(s1);
int y = getGender(s2);
way[x][y] = 1;
way[y][x] = 1;
hv[x] = 1; hv[y] = 1;
}
int cnt = 1;
for (int i = 0; i <= 9999; ++i) if (hv[i]) id[cnt++] = i;
scanf("%d",&K);
for (int i = 0; i < K; ++i) {
int x, y;
scanf("%d%d", &x, &y);
int ans = find(abs(x), abs(y));
printf("%d\n", ans);
for (int j = 1; j <= ans; ++j)
printf("%04d %04d\n",lst[j][0],lst[j][1]);
}
return 0;
}