算法提高之LeetCode刷题数据结构和算法分析

290. 单词模式(Python)

2019-05-16  本文已影响1人  玖月晴

题目

难度:★★☆☆☆
类型:字符串

给定一种 pattern(模式) 和一个字符串 str ,判断 str 是否遵循相同的模式。

这里的遵循指完全匹配,例如, pattern 里的每个字母和字符串 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

解答

这道题与【题目205. 同构字符串】十分相似,我们同样采用其中的解法,当满足下列条件时,两者匹配:

模式字符去重后的长度 == 句子中单词去重后的长度 == 模式-单词对去重后的长度

这里需要注意的是,输入的模式长度与句子中单词个数存在不相等的情况,这样一定无法配对。

class Solution(object):
    def wordPattern(self, pattern, str):
        """
        :type pattern: str
        :type str: str
        :rtype: bool
        """

        str_list = str.split(' ')                               # 将句子用空格分开组成单词列表
        if len(pattern) == len(str_list):                       # 如果模式个数等于单词个数
            pattern_str_pairs = zip(pattern, str.split(' '))    # 打包组成模式-单词对
            # 只有模式集合长度、单词集合长度与模式-单词对集合长度相等时,才能匹配
            return len(set(pattern)) == len(set(str_list)) == len(set(pattern_str_pairs))
        else:                                                   # 如果模式个数与单词个数不同
            return False                                        # 一定无法匹配

下面是更加紧凑的写法,一行代码实现,和以上代码道理完全相同。

class Solution(object):
    def wordPattern(self, pattern, str):
        """
        :type pattern: str
        :type str: str
        :rtype: bool
        """
 
        return len(set(pattern)) == len(set(str.split(' '))) == len(set(zip(pattern, str.split(' ')))) if len(pattern) == len(str.split(' ')) else False

如有疑问或建议,欢迎评论区留言~

上一篇下一篇

猜你喜欢

热点阅读