leetcode 966. 元音拼写检查器 map解法

在给定单词列表 wordlist 的情况下，我们希望实现一个拼写检查器，将查询单词转换为正确的单词。

对于给定的查询单词 query，拼写检查器将会处理两类拼写错误：

• 大小写：如果查询匹配单词列表中的某个单词（不区分大小写），则返回的正确单词与单词列表中的大小写相同。
  • 例如：wordlist =[“yellow”], query = “YellOw”: correct = “yellow”
  • 例如：wordlist = [“Yellow”], query = “yellow”: correct = “Yellow”
  • 例如：wordlist = [“yellow”], query = “yellow”: correct = “yellow”
• 元音错误：如果在将查询单词中的元音（‘a’、‘e’、‘i’、‘o’、‘u’）分别替换为任何元音后，能与单词列表中的单词匹配（不区分大小写），则返回的正确单词与单词列表中的匹配项大小写相同。
  • 例如：wordlist = [“YellOw”], query = “yollow”: correct = “YellOw”
  • 例如：wordlist = [“YellOw”], query = “yeellow”: correct = “” （无匹配项）
  • 例如：wordlist = [“YellOw”], query = “yllw”: correct = “” （无匹配项）

此外，拼写检查器还按照以下优先级规则操作：

• 当查询完全匹配单词列表中的某个单词（区分大小写）时，应返回相同的单词。
• 当查询匹配到大小写问题的单词时，您应该返回单词列表中的第一个这样的匹配项。
• 当查询匹配到元音错误的单词时，您应该返回单词列表中的第一个这样的匹配项。
• 如果该查询在单词列表中没有匹配项，则应返回空字符串。

给出一些查询 queries，返回一个单词答案列表 answer，其中 answer[i] 是由查询 query = queries[i] 得到的正确单词。

示例：

输入：wordlist = [“KiTe”,“kite”,“hare”,“Hare”], queries = [“kite”,“Kite”,“KiTe”,“Hare”,“HARE”,“Hear”,“hear”,“keti”,“keet”,“keto”]
输出：[“kite”,“KiTe”,“KiTe”,“Hare”,“hare”,"","",“KiTe”,"",“KiTe”]

提示：

1 <= wordlist.length <= 5000
1 <= queries.length <= 5000
1 <=> wordlist[i].length <= 7
1 <= queries[i].length <= 7
wordlist 和 queries 中的所有字符串仅由英文字母组成。

拿到题，排出匹配的优先级。
首先，不管大小写，只要辅音一样就算匹配，否则得到空串。比如，Koto和字典KiTe匹配。
之后，优先匹配完全一致的，例如kite和字典kite匹配。再匹配字母一样仅大小写不同的，例如kite和字典KiTe匹配。最后，按照给出的字典顺序匹配第一个辅音一致的。

这样，建立一个map，key为字典转小写，再把元音替换掉的结果。比如KiTe的key为k.t.。这样，对每一个query进行同样操作，与key匹配，则表明一定可以得到一个结果。
map的value分两部分，第1部分存储符合key的wordlist原文，例如KiTe, kiTe, Keto，这部分按照读入顺序存储。第2部分储存符合key的wordlist的小写形式，例如kite, kite, keto。

class Solution {
public:
string stolower(const string &s)//转小写
{
	string sr = s;
	for (auto &c : sr)
		c = tolower(c);
	return sr;
}

string   replace(const string &s)//s中的元音替换为.
{
	string sr = s;
	for (auto &c : sr)
	{
		if (c == 'a' || c == 'e' || c == 'i' || c == 'o' || c == 'u')
			c = '.';
	}
	return sr;
}

vector<string> spellchecker(vector<string>& wordlist, vector<string>& queries) {
	map<string, pair<vector<string>, vector<string>>> mm;
	for (auto &s : wordlist)
	{
		string sa = replace(stolower(s));//key为s转小写，再替换元音的结果
		mm[sa].first.push_back(s);//value第1部分存原文
		mm[sa].second.push_back(stolower(s));//value第2部分存小写形式
	}

	vector<string> result;
	for (auto &s : queries)
	{
		auto it = mm.find(replace(stolower(s)));
		if (it == mm.end())//map中没有辅音一致的结果
			result.push_back("");
		else
		{
			auto matchall = find(it->second.first.begin(), it->second.first.end(), s);//s在map的第1部分有全文匹配项
			if (matchall != it->second.first.end())
			{
				result.push_back(s);
			}
			else
			{
				auto matchlow = find(it->second.second.begin(), it->second.second.end(), stolower(s));//s在map的第2部分中有小写匹配项
				if (matchlow != it->second.second.end())
				{
					int index = matchlow-it->second.second.begin() ;
					string r = *(it->second.first.begin() + index);
					result.push_back(r);
				}
				else
				{//s既不匹配全文，也不匹配小写，说明s仅仅是辅音匹配。这样返回wordlist中第一个和s辅音匹配的项。因为map的第1部分是按照wordlist顺序存储的，所以直接取第1个即可。
					result.push_back(it->second.first.front());
				}
			}
		}
	}
	return result;
}
};