天天看点

leecode 890 ~查找和替换模式

查找和替换模式

你有一个单词列表 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 static List<String> findAndReplacePattern(String[] words, String pattern) {
        int[] p = F(pattern);
        List<String> res = new ArrayList<String>();

        for (String word : words)
            if (Arrays.equals(F(word), p))
                res.add(word);
        return res;
    }

    public static int[] F(String word) {
        HashMap<Character, Integer> map = new HashMap<>();
        int wordLen = word.length();
        int[] res = new int[wordLen];

        for (int i = ; i < wordLen; i++) {
            map.putIfAbsent(word.charAt(i), map.size());
            res[i] = map.get(word.charAt(i));
        }
        return res;
    }

    public static void main(String[] args) {
        String[] words={"abc","deq","mee","aqq","dkd","ccc"};
        String pattern="abb";
        System.out.println(findAndReplacePattern(words,pattern));
    }
}