890. find and replace mode

Original link ：890. Find and replace patterns

 

Solution:

          Use two arrays , When traversing a string ,words[i] Store word to pattern string mappings ,pattern[i] Storage mode string to word mapping .

class Solution {
public:
    vector<string> findAndReplacePattern(vector<string>& words, string pattern) {
        int n = words.size();
        vector<string> res; // Return value 
        for(int i = 0;i < n;i++) {
            if(check(words[i], pattern)) res.push_back(words[i]);
        }
        return res;
    }

    bool check(const string &str,const string &patt) {
        int words[26],pattern[26];
        memset(words, -1, sizeof(words));
        memset(pattern, -1, sizeof(pattern));
        
        for(int i = 0;i < str.size();i++) {
            int st = str[i] - 'a';
            int ts = patt[i] - 'a';
            
            if(words[st] == -1 && pattern[ts] == -1) {  // If the letters are not mapped , Store the mapping relationship between letters 
                words[st] = ts;
                pattern[ts] = st;   
            }
            else if(words[st] != ts || pattern[ts] != st) {  // Determine whether the mapping relationship is correct 
                return false;
            } 
        }
        return true;
    }
};