PAT (Advanced Level) 1139——不使用map但还是踩了点小坑

题目传送门

• 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;
}