哈希表题目：单词规律

题目

标题和出处

标题：单词规律

出处：290. 单词规律

难度

3 级

题目描述

要求

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

这里的遵循指完全匹配，$\text{pattern}$ 中的每个字母和 $\text{s}$ 中的每个非空单词之间存在双射关系。

示例

示例 1：

输入：$\text{pattern = "abba", s = "dog cat cat dog"}$
输出：$\text{true}$

示例 2：

输入：$\text{pattern = "abba", s = "dog cat cat fish"}$
输出：$\text{false}$

示例 3：

输入：$\text{pattern = "aaaa", s = "dog cat cat dog"}$
输出：$\text{false}$

示例 4：

输入：$\text{pattern = "abba", s = "dog dog dog dog"}$
输出：$\text{false}$

数据范围

• $\text{1}\le \text{pattern.length}\le \text{300}$
• $\text{pattern}$ 只包含小写英语字母
• $\text{1}\le \text{s.length}\le \text{3000}$
• $\text{s}$ 只包含小写英语字母和空格 $\text{‘ ’}$
• $\text{s}$ 不包含开头和结尾的空格
• $\text{s}$ 中的所有单词由一个空格分隔

解法

思路和算法

如果字符串 $s$ 遵循相同的规律，则模式 $\text{pattern}$ 中的字母和字符串 $s$ 中的单词之间的关系是双射关系，即模式 $\text{pattern}$ 中的每个字母都对应字符串 $s$ 中的一个单词，字符串 $s$ 中的每个单词都对应模式 $\text{pattern}$ 中的一个字母。

由于字符串 $s$ 由单词组成，因此首先将字符串 $s$ 转化成字符串数组。如果字符串数组的长度和 $\text{pattern}$ 的长度不同，则无法形成双射关系，返回 $\text{false}$。

判断是否满足双射关系，可以使用两个哈希表分别存储 $\text{pattern}$ 到 $s$ 的映射和 $s$ 到 $\text{pattern}$ 的映射。

同时遍历 $\text{pattern}$ 和 $s$ 转化成的字符串数组，对于相同位置的字符 $c$ 和单词 $\text{word}$，分别判断 $c$ 和 $\text{word}$ 是否在两个哈希表中存在，如果不存在则将对应的关系存入哈希表，如果已经存在则判断哈希表中对应的关系和当前的关系是否相同，如果不相同则不满足双射关系，返回 $\text{false}$。

如果遍历结束之后没有发现不满足双射关系的情况，则返回 $\text{true}$。

代码

class Solution {
    public boolean wordPattern(String pattern, String s) {
        Map<Character, String> map1 = new HashMap<Character, String>();
        Map<String, Character> map2 = new HashMap<String, Character>();
        int length = pattern.length();
        String[] arr = s.split(" ");
        if (arr.length != length) {
            return false;
        }
        for (int i = 0; i < length; i++) {
            char c = pattern.charAt(i);
            String word = arr[i];
            if (!map1.containsKey(c)) {
                map1.put(c, word);
            } else if (!map1.get(c).equals(word)) {
                return false;
            }
            if (!map2.containsKey(word)) {
                map2.put(word, c);
            } else if (map2.get(word) != c) {
                return false;
            }
        }
        return true;
    }
}

复杂度分析

• 时间复杂度：$O(m+n)$，其中 $m$ 和 $n$ 分别是字符串 $\text{pattern}$ 和 $s$ 的长度。需要遍历字符串 $\text{pattern}$ 和 $s$ 各一次。

• 空间复杂度：$O(m+n)$，其中 $m$ 和 $n$ 分别是字符串 $\text{pattern}$ 和 $s$ 的长度。空间复杂度主要取决于哈希表存储的对应关系。