你有一个单词列表 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
解析:写函数,return 数组,abb就是1 2 2。所有的words和pattern通过函数后比较下;
代码如下:
class Solution {
public:
vector<string> findAndReplacePattern(vector<string>& words, string pattern) {
vector<string> res;
int len=pattern.length();
vector<int> cpr=restruct(pattern);
for(vector<string>::iterator it=words.begin();it!=words.end();it++){
if(restruct(*it)==cpr) res.push_back(*it);
}
return res;
}
vector<int> restruct(string s) {
map<int, char>a;
vector<int>res;
int biggest = 0;
for (int i = 0; i<s.length(); i++) {
if (a.count(s[i])) res.push_back(a[s[i]]);
else {
a[s[i]] = biggest; res.push_back(a[s[i]]); biggest++;
}
}
return res;
}
};