你有一个单词列表 words 和一个模式 pattern,你想知道 words 中的哪些单词与模式匹配。
如果存在字母的排列 p ,使得将模式中的每个字母 x 替换为 p(x) 之后,我们就得到了所需的单词,那么单词与模式是匹配的。
(回想一下,字母的排列是从字母到字母的双射:每个字母映射到另一个字母,没有两个字母映射到同一个字母。)
返回 words 中与给定模式匹配的单词列表。
你可以按任何顺序返回答案。
示例:
输入:words = ["abc","deq","mee","aqq","dkd","ccc"], pattern = "abb"
输出:["mee","aqq"]
解释:
"mee" 与模式匹配,因为存在排列 {a -> m, b -> e, ...}。
"ccc" 与模式不匹配,因为 {a -> c, b -> c, ...} 不是排列。
因为 a 和 b 映射到同一个字母。
提示:
1 <= words.length <= 50
1 <= pattern.length = words[i].length <= 20
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/find-and-replace-pattern 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public:
vector<string> findAndReplacePattern(vector<string>& words, string pattern) {
unordered_map<char,int> m;//pattern字符,pattern与word的字符差
unordered_map<char,int> n;//word字符,pattern与word的字符差
vector<string> ans;
int i;
bool good;
for(string &word : words)
{
if(word.size() != pattern.size())
continue;
m.clear();
n.clear();
good = true;
for(i = 0; i < pattern.size(); ++i)
{
if(m.find(pattern[i]) == m.end())//没找到,存入字符差值
m[pattern[i]] = word[i]-pattern[i];
else
{
if(m[pattern[i]] != word[i]-pattern[i])
{ //找到了,之前存过,跟之前的差值一致吗?不一致,错
good = false;
break;
}
}
if(n.find(word[i]) == n.end())//没找到,存入字符差值
n[word[i]] = word[i]-pattern[i];
else
{
if(n[word[i]] != word[i]-pattern[i])
{ //找到了,之前存过,跟之前的差值一致吗?不一致,错
good = false;
break;
}
}
}
if(good)
ans.push_back(word);
}
return ans;
}
};