为了账号安全,请及时绑定邮箱和手机立即绑定

leetcode每日一题:290. 单词规律

标签:
Python

一起刷题吧

一、题意分析

输入:两个字符串
输出:如果第二个字符串与第一个字符串模式相同,则返回 true,否则返回 false
难度:简单
标签:哈希

示例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

二、参考代码

这个题目就很简单啦,直接使用哈希表存储两者的对应关系就可以了,需要注意的是,需要使用两个哈希来存储对应关系,用一个哈希表会有问题,比如反例:

print(s.wordPattern("abba", "dog cat cat fish"))
print(s.wordPattern("abba", "dog dog dog dog"))

我的实现代码:

class Solution:
    def wordPattern(self, pattern: str, s: str) -> bool:
        if not pattern:
            return not s
        if not s:
            return not pattern
        i = 0
        s = s.split(" ")
        if len(s) != len(pattern):
            return False
        mapping = {}
        rev = set()
        while i < len(pattern):
            if pattern[i] not in mapping:
                if s[i] not in rev:
                    mapping[pattern[i]] = s[i]
                    rev.add(s[i])
                else:
                    return False
            else:
                if mapping[pattern[i]] != s[i]:
                    return False
            i += 1
        return True

参考了下官方给的 python 代码示例,代码写得更加优雅,代码如下:

class Solution:
    def wordPattern(self, pattern: str, s: str) -> bool:
        word2ch = dict()
        ch2word = dict()
        words = s.split()
        if len(pattern) != len(words):
            return False
        
        for ch, word in zip(pattern, words):
            # 更简洁
            if (word in word2ch and word2ch[word] != ch) or (ch in ch2word and ch2word[ch] != word):
                return False
            word2ch[word] = ch
            ch2word[ch] = word
    
        return True

# 作者:LeetCode-Solution
# 链接:https://leetcode-cn.com/problems/word-pattern/solution/dan-ci-gui-lu-by-leetcode-solution-6vqv/
# 来源:力扣(LeetCode)
# 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

点击查看更多内容
TA 点赞

若觉得本文不错,就分享一下吧!

评论

作者其他优质文章

正在加载中
  • 推荐
  • 评论
  • 收藏
  • 共同学习,写下你的评论
感谢您的支持,我会继续努力的~
扫码打赏,你说多少就多少
赞赏金额会直接到老师账户
支付方式
打开微信扫一扫,即可进行扫码打赏哦
今天注册有机会得

100积分直接送

付费专栏免费学

大额优惠券免费领

立即参与 放弃机会
意见反馈 帮助中心 APP下载
官方微信

举报

0/150
提交
取消