A 1139 First Contact (30分)

Posted 2020-12-11 tsruixi

一、技术总结

1. 学会使用unordered_map,可以减少存储空间，同时如果是两个id号之间有关系，可以直接让他们拼接起来。使用map<int, bool>来表示他们直接是否有关系；
2. 学会使用stoi函数，能够将字符串直接转化为默认是十进制的数字。
3. 最后就是读懂题目,如果A喜欢B，则需要寻找与A同性别的朋友C，和与B同性别的朋友D，如果C和D是朋友，那么就可以输出结果。同时注意输出的顺序，对于C按升序输出，如果C相同那么则按D的升序输出。
4. 题目中还有一条要求是，如果A与B直接是朋友关系则不算入；
5. 存储同性朋友，使用vector，表示朋友关系的使用arr变量，也就第一点提到的类型进行存储。

二、参考代码

#include<iostream>
#include<vector> 
#include<string>
#include<algorithm>
#include<unordered_map>
using namespace std;
unordered_map<int, bool> arr;
struct node{
	int a, b;
};
bool cmp(node x, node y){
	return x.a != y.a ? x.a < y.a : x.b < y.b; 
}
int main(){
	int n, m, k;
	scanf("%d%d", &n, &m);
	vector<int> v[10000];
	for(int i = 0; i < m; i++){
		string a, b;
		cin >> a >> b;
		if(a.length() == b.length()){
			v[abs(stoi(a))].push_back(abs(stoi(b)));
			v[abs(stoi(b))].push_back(abs(stoi(a)));
		}
		arr[abs(stoi(a)) * 10000 + abs(stoi(b))] = arr[abs(stoi(b)) * 10000 + abs(stoi(a))] = true;
	}
	scanf("%d", &k);
	for(int i = 0; i < k; i++){
		int c, d;
		cin >> c >> d;
		vector<node> ans;
		for(int j = 0; j < v[abs(c)].size(); j++){
			for(int k = 0; k < v[abs(d)].size(); k++){
				if(v[abs(c)][j] == abs(d) || v[abs(d)][k] == abs(c)) continue;
				if(arr[v[abs(c)][j] * 10000 + v[abs(d)][k]] == true){
					ans.push_back(node{v[abs(c)][j], v[abs(d)][k]});
				}
			}
		}
		sort(ans.begin(), ans.end(), cmp);
		printf("%d
", int(ans.size()));
		for(int j = 0; j < ans.size(); j++){
			printf("%04d %04d
", ans[j].a, ans[j].b);
		}	
	}
	return 0;
}