PAT甲级1139 First Contact (30 分)：[C++题解] 图论、暴力枚举两个点、hash映射

文章目录

题目分析
题目链接

题目分析

来源：acwing

题目分析：

图论模拟题。

给定暗恋的两个人A 和B，需要寻找一对C 和D ，满足:A和C是朋友，C和D是朋友，D和B是朋友。而且A、C同性别，B、D同性别。题目给的暗恋：可能是异性(男对女，或者女对男)，也可能是同性暗恋同性。

数据不大，枚举即可。只要建好图标记好边g[N][N]，然后直接暴力枚举即可

for(auto c :a的同性别阵营 )for(auto d : b的同性别阵营)if( c != a && c != b && d != a && d != b && g[a][c] && g[c][d] && g[d][b] )res.push_back({c , d});

题目的难点之一在于给定的不是数字，而是字符串，因为0000 和 -0000表示男生和女生，不能单纯从数字上来看，所以用字符串来存，这样的话，需要做一个从字符串到数字的映射，需要使用hash表。输出的时候还需要输出字符串（不需要输出表示女生的负号），因此还需要数字和字符串的逆映射，需要不停地处理输入输出。

第二个难点是枚举的时候需要按照性别枚举，这时候男生和女生需要分别存。

STL语法点补充：
下面使用了vector的去重，使用方法erase() 和uniue().

vector<int> boys;
boys.erase(  unique(boys.begin(),boys.end()),   boys.end() );

上面的代码是对数组boys进行删除重复元素。

还用到了STL中的hash table unordered_map<string,int > mp;实现从字符串到数字的映射。使用方法count()判断是否已经映射过。如果mp.count(x) == 0表示还没有映射过。

ac代码

#include<bits/stdc++.h>
using namespace std;
const int N =  310;int n, m;
bool g[N][N];
string num[N];
unordered_map<string,int > mp;
int id ;
vector<int> boys, girls;
int main(){cin >> n  >> m;while(m--){string  a, b;cin >> a >> b;string x =a ,y =b;if(x.size()==5) x =x.substr(1);if(y.size() ==5) y = y .substr(1);//没出现过，映射成全新的数if(mp.count(x) == 0) mp[x] = ++id,num[id] = x;//原来的编号是什么if(mp.count(y) == 0) mp[y] = ++id, num[id] = y;int px= mp[x] ,py = mp[y];//之间有边g[px][py] = g[py][px] = true;if(a[0] != '-') boys. push_back(px);else girls.push_back(px);if(b[0] != '-' ) boys. push_back(py);else girls.push_back(py);}sort(boys.begin(),boys.end());//删除重复元素boys.erase(unique(boys.begin(),boys.end()), boys.end());sort(girls.begin(),girls.end());//删除重复元素girls.erase(unique(girls.begin(),girls.end()), girls.end());int k ;cin >> k;//询问while(k--){vector<pair<string,string>> res; //存结果string  x, y;cin >> x   >> y;//判断是男孩还是女孩vector<int> p= boys, q = boys;if(x[0] == '-') p =girls, x = x. substr(1);if( y[0] == '-') q = girls, y = y.substr(1);int a = mp[x] ,b = mp[y];//图的遍历//找两个点for(int c: p)  //c从a的性别里找for(int d : q){ //d从b的性别里找if( c !=a && c !=b  &&  d!= a && d!= b && g[a][c] && g[c][d] && g[d][b])res.push_back({num[c],num[d]});}cout<< res.size()<<endl;sort(res.begin(),res.end());for( auto p :res ) cout << p.first <<" "<< p.second <<endl;}}

题目链接

PAT甲级1139 First Contact (30 分)