力扣——查找和替换模式

Posted 2021-02-12 jaypark

你有一个单词列表 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

 

class Solution {
    public List<String> findAndReplacePattern(String[] words, String pattern) {
        List<String> res = new ArrayList<>(words.length);
        for (String word : words) {
            if (likePattern(word, pattern)) {
                res.add(word);
            }
        }
        return res;
    }

    public boolean likePattern(String word, String pattern) {
        if (word.length() != pattern.length()) {
            return false;
        }
        char[] wordChars = word.toCharArray();
        boolean[] flagOfLetter = new boolean[26];
        boolean[] flagOfWord = new boolean[wordChars.length];
        for (int i = 0; i < word.length(); i++) {
            int indexOfLetter = pattern.charAt(i) - ‘a‘;
            if (!flagOfWord[i] && !flagOfLetter[indexOfLetter]) {
                char k = wordChars[i];
                flagOfLetter[indexOfLetter] = true;
                // 如果这个字母已经被转化过，那么结束.
                for (int j = i; j < word.length(); j++) {
                    if (wordChars[j] == k && !flagOfWord[j]) {
                        flagOfWord[j] = true;
                        wordChars[j] = pattern.charAt(i);
                    }
                }
            }
        }
        for (int i = 0; i < word.length(); i++) {
            if (wordChars[i] != pattern.charAt(i)) {
                return false;
            }
        }
        return true;

    }
}