单词规律

开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第19天，点击查看活动详情

290. 单词规律 - 力扣（LeetCode）

给定一种规律 pattern 和一个字符串 s ，判断 s 是否遵循相同的规律。

这里的 遵循 指完全匹配，例如， pattern 里的每个字母和字符串 s 中的每个非空单词之间存在着双向连接的对应规律。

示例1:

输入: pattern = "abba", s = "dog cat cat dog"
输出: true

示例 2:

输入: pattern = "abba", s = "dog cat cat fish"
输出: false

示例 3:

输入: pattern = "aaaa", s = "dog cat cat dog"
输出: false

提示:

• 1 <= pattern.length <= 300
• pattern 只包含小写英文字母
• 1 <= s.length <= 3000
• s 只包含小写英文字母和 ' '
• s 不包含 任何前导或尾随对空格
• s 中每个单词都被 单个空格 分隔

思路

设由s拆分出来的单词表为words，由题目知完全匹配指的是当words的中的一个单词words[i]，如果这个单词在他之前的j位置也出现过，即words[i] === word[j]，则pattern[i] === pattern[j]，如果单词在他之前都没有出现过，则pattern[i]在pattern也应该是首次出现。

顺着上面的思路，我们可以用双层循环遍历来求解，外层遍历words，记当前下标为i，内层遍历0到i-1，记当前下标为j，如果(words[i] === words[j]) && (pattern[i] === pattern[j])或者(words[i] !== words[j]) && (pattern[i] !== pattern[j])继续遍历，否则表示单词和规律不匹配，跳出循环，返回false，遍历完成后表示表示单词和规律匹配，返回true。

上面步骤比较麻烦的是内层循环查找比较的过程，我们可以用两个哈希表来优化。我们用mapA来存储单词和规律的映射关系，用mapB来存储规律和单词的映射关系。遍历words，如果mapA中存在key为words[i]，比较mapA[words[i]]和pattern[i]，如果相等继续，如果不相等，表示不匹配，如果mapA中不存在key为word[i],把word[i]和pattern[i]的映射关系添加到mapA。以同样方式处理mapB。

解题

/**
 * @param {string} pattern
 * @param {string} s
 * @return {boolean}
 */
var wordPattern = function (pattern, s) {
  const words = s.split(" ");
  if (words.length !== pattern.length) return false;
  const match = (pattern, s) => {
    const map = new Map();
    for (let i = 0; i < pattern.length; i++) {
      if (map.has(pattern[i])) {
        if (map.get(pattern[i]) !== s[i]) {
          return false;
        }
      } else {
        map.set(pattern[i], s[i]);
      }
    }
    return true;
  };
  return match(pattern, words) && match(words, pattern)
};