Skip to content

Latest commit

 

History

History
61 lines (47 loc) · 1.63 KB

File metadata and controls

61 lines (47 loc) · 1.63 KB

查找和替换模式

难度:中等

https://leetcode.cn/problems/find-and-replace-pattern/

题目

你有一个单词列表 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 映射到同一个字母。

解题

/**
 * 构建双射
 * @desc 时间复杂度 O(MN)  空间复杂度 O(N)
 * @param words
 * @param pattern
 * @returns
 */
export function findAndReplacePattern(words: string[], pattern: string): string[] {
  const ans: string[] = []
  for (const word of words)
    if (match(word, pattern) && match(pattern, word)) ans.push(word)
  return ans

  function match(word: string, pattern: string) {
    const map = new Map<string, string>()
    for (let i = 0; i < word.length; i++) {
      const x = word[i]
      const y = pattern[i]
      if (!map.has(x))
        map.set(x, y)

      else if (map.get(x) !== y)
        return false
    }

    return true
  }
}