LeetCode No890. 查找和替换模式

题目

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

思路

题目给的提示已经很明显了，用两个map构建映射关系，如果两个映射关系都没有冲突，则说明是匹配的。

AC代码

点击查看代码

class Solution {

    private boolean isOk(String a, String b) {
        if( a.length()!=b.length() ) {
            return false;
        }
        char[] as = a.toCharArray();
        char[] bs = b.toCharArray();
        int len = as.length;
        HashMap<Character, Character> map = new HashMap<>();
        HashMap<Character, Character> map2 = new HashMap<>();
        for(int i=0; i<len; i++) {
            Character ch = map.get(bs[i]);
            Character ch2 = map2.get(as[i]);
            if( ch2!=null && ch2!=bs[i] ) {
                return false;
            }
            if( ch==null ) {
                map.put(bs[i], as[i]);
                map2.put(as[i], bs[i]);
                continue;
            }
            if( ch!=as[i]) {
                return false;
            }
        }
        return true;
    }

    public List<String> findAndReplacePattern(String[] words, String pattern) {
        List<String> ans = new ArrayList<>();
        for(String str: words ) {
            if( isOk(str, pattern) ) {
                ans.add(str);
            }
        }
        return ans;
    }
}