aboutsummaryrefslogtreecommitdiff
path: root/find_replace_pattern.cpp
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);
  }
};