题目
给定一种规律 pattern 和一个字符串 str ,判断 str 是否遵循相同的规律。
这里的 遵循 指完全匹配,例如, pattern 里的每个字母和字符串 str 中的每个非空单词之间存在着双向连接的对应规律。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/word-pattern
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
示例
示例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 包含了由单个空格分隔的小写字母。
解答
先将两个输入的字符串都转成列表,判断两个列表的长度,不相等直接返回False。
再将两个列表去重判断长度是否相等,不相等直接返回False。
然后遍历列表,
判断如果列表A的当前元素和后面一位元素相同但是列表B的当前元素和后面一位元素不相同,返回False;
如果列表A的当前元素和后面一位元素不相同但是列表B的当前元素和后面一位元素相同,返回False。
最后如果都没有返回False,那么两个列表规则匹配,返回True。
class Solution:
def wordPattern(self, pattern, s):
pattern = list(pattern)
s = s.split(' ')
if len(pattern) != len(s):
return False
if len(set(pattern)) != len(set(s)):
return False
for i in range(len(pattern) - 1):
if pattern[i] == pattern[i + 1] and s[i] != s[i + 1]:
return False
if pattern[i] != pattern[i + 1] and s[i] == s[i + 1]:
return False
return True
if __name__ == '__main__':
pattern = "abca"
str = "dog cat cat dog"
res = Solution().wordPattern(pattern, str)
print(res)