L2-050 懂蛇语(2024PTA天梯赛)

L2-050 懂蛇语

L2-050 懂蛇语 - 团体程序设计天梯赛-练习集 (pintia.cn)
前置知识: [[字符串流式处理#字符串分割]]
核心在于用map储存 尤其是map的值用vector来储存多个字符串s

#include <bits/stdc++.h>
using namespace std;

map<string, vector<string> >mp; //用vector<string>是因为适应句子不唯一的情况
string s, temp;

int main()
{
	int n;
	cin >> n;
	cin.ignore();
	while (n--)
	{
		string specialS; //specialS要单独的放在while循环里 因为ss>> 要后继的temp为空才好操作
		getline(cin, s);
		stringstream ss(s);
		while (ss >> temp)
		{
			specialS += temp[0]; //将每个单词的首字母缩写放到specialS里面
		}
		mp[specialS].push_back(s);
	}
	int query;
	cin >> query;
	cin.ignore();
	while (query--)
	{
		string specialS; //specialS要单独的放在while循环里 因为ss>> 要后继的temp为空才好操作
		getline(cin, s);
		stringstream ss(s);
		while (ss >> temp)
		{
			specialS += temp[0];
		}
		if (mp.count(specialS))
		{
			vector<string>ans = mp[specialS]; // 将mp中键为specialS的值vector<string>赋值给ans数组
			sort(ans.begin(), ans.end()); // 如果句子不唯一，则按整句的字母序输出 sort的操作对象为string时 遵守的规则是字典序
			cout << ans[0];
			for (int i = 1; i < ans.size(); i++)
			{
				cout << "|" << ans[i];
			}
			cout << endl; //末尾换行 对应多行输入输出
		}
		else cout << s << endl;
	}
	return 0;
}