LeetCode290.单词规律Golang版

LeetCode290.单词规律Golang版

1. 问题描述

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

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

示例1:

输入: pattern = “abba”, str = “dog cat cat dog”
输出: true
示例 2:

输入:pattern = “abba”, str = “dog cat cat fish”
输出: false
示例 3:

输入: pattern = “aaaa”, str = “dog cat cat dog”
输出: false
示例 4:

输入: pattern = “abba”, str = “dog dog dog dog”
输出: false
说明:
你可以假设 pattern 只包含小写字母， str 包含了由单个空格分隔的小写字母。

2. 思路

1. 维护两个map[byte]byte类型的map,key -> pattern[i] value -> s[i]

注意这是一个双射关系
word2ch[i] == s[i]，同时也要满足 ch2word[j] == s[j]

3. 遍历s,一旦发现key和value不对应，返回false

3. 代码

func wordPattern(pattern string, s string) bool {

    word2ch := map[string]byte{}
    ch2word := map[byte]string{}
    words := strings.Split(s," ")

    if len(words) != len(pattern) {
        return false
    }

    for i := 0; i < len(words); i++ {
        if word2ch[words[i]] > 0 && word2ch[words[i]] != pattern[i] || ch2word[pattern[i]] != "" && ch2word[pattern[i]] != words[i] {
            return false
        }
        word2ch[words[i]] = pattern[i]
        ch2word[pattern[i]] = words[i]
    }
    return true
}