当前位置: 代码迷 >> 综合 >> PAT1139 First Contact(模拟)
  详细解决方案

PAT1139 First Contact(模拟)

热度:1   发布时间:2024-01-16 13:17:48.0

题意:

A和B告白,必须通过A的同性朋友和B的同性朋友传达,要求输出所以可能的两个朋友的情况

思路:

这题有些复杂,我一开始想复杂了,上来就用dfs+回溯做,结果超时,其实只要遍历所有A的朋友和B的朋友,看这两者是否互为朋友即可。不过这题还有很多地方要注意,我还是看别人的代码写的。

https://www.liuchuo.net/archives/4210

#include<iostream>
#include<algorithm>
#include<vector>
#include<functional>
#include<string>
#include<cmath>
#include<queue>
#include<map>
#include<unordered_map>
#include<set>
using namespace std;
const int maxn = 10050;
const int inf = 10000;
unordered_map<int, bool> arr;
vector<int> v[maxn];
struct node {int a, b;
};bool cmp(node x, node y) {if (x.a != y.a) {return x.a < y.a;} else {return x.b < y.b;}
}int main() {int n, m, k;scanf("%d%d", &n, &m);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))* inf + abs(stoi(b))] = arr[abs(stoi(b))* inf+ 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 l = 0; l < v[abs(d)].size(); l++) {int s = v[abs(c)][j];int e = v[abs(d)][l];if (s == abs(d) || e == abs(c))continue;if (arr[s*inf + e] == true) {ans.push_back(node{ s,e });}}}sort(ans.begin(), ans.end(), cmp);printf("%d\n", ans.size());for (int j = 0; j < ans.size(); j++) {printf("%04d %04d\n", ans[j].a, ans[j].b);}}return 0;
}

 

  相关解决方案