blob: a00bfcb6352b9971e06b2f776fe285458ae15905 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
#include "leetcode.h"
class Solution {
public:
vector<string> findAndReplacePattern(vector<string> &words, string pattern) {
ans.resize(0);
codex.clear();
cipher = pattern;
for (int i = 0; i < pattern.size(); i++)
cipher[i] = translate(cipher[i]);
for (auto &word : words)
compare(word);
return ans;
}
private:
vector<string> ans;
unordered_map<char, char> codex;
string cipher;
char translate(char &c) {
if (codex.find(c) == codex.end())
codex[c] = (char)(97 + codex.size());
return codex[c];
}
void compare(string &word) {
codex.clear();
for (int i = 0; i < word.size(); i++)
if (translate(word[i]) != cipher[i])
return;
ans.push_back(word);
}
};
|