力扣——查找和替换模式
你有一个单词列表 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; } }