携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第11天,点击查看活动详情
一、题目描述:
给定一种规律 pattern 和一个字符串 s ,判断 s 是否遵循相同的规律。
这里的 遵循 指完全匹配,例如, 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
提示:
- 1 <= pattern.length <= 300
- pattern 只包含小写英文字母
- 1 <= s.length <= 3000
- s 只包含小写英文字母和 ' '
- s 不包含 任何前导或尾随对空格
- s 中每个单词都被 单个空格 分隔
二、思路分析:
思路很简单
- 将字符串pattern和s转换为list,再将其转为二维矩阵即可。
这一步需要注意pattern和s列表应该是等长的,否则不能转成二维矩阵,第19个测试用例pat="aaa",s="aa aa aa aa"就不等长,导致二维矩阵res_list = [['a', 'aa'], ['a', 'aa'], ['a', 'aa']]缺失了s列表的最后一个"aa"。
本来我是想用字典(哈希表)来代替二维矩阵进行存储的,但是我发现字典和set(集合)有点像,会自动合并键值对相同的元素,这一点很烦人,诸如pat="ss",s="ab ab",则res_dict = {'s': 'ab'},自动合并了两个相同的键值对 - 对二维矩阵进行双指针操作即可,2次for循环,时间复杂度O(n^2)。
三、AC 代码:
''' 输入: pattern = "abba", str = "dog cat cat dog"
输出: true
输入:pattern = "abba", str = "dog cat cat fish"
输出: false'''
class Solution:
def wordPattern(self, pattern: str, s: str) -> bool:
pattern = list(pattern)
s = list(s.split(' ', len(s)-1))
if len(pattern) != len(s): return False # 防止第19个用例
# print(pattern, s) # ['a', 'b', 'b', 'a'] ['dog', 'cat', 'cat', 'dog']
res_list = [[pattern[i], s[i]] for i in range(len(pattern))]
# print(res_list) # [['a', 'dog'], ['b', 'cat'], ['b', 'cat'], ['a', 'dog']]
for i in range(len(res_list)-1):
for j in range(i+1, len(res_list)):
# print(f'{i}:{j}') # 0:1 0:2 0:3 1:2 1:3 2:3
if res_list[i][0] != res_list[j][0] and res_list[i][1] == res_list[j][1]:
return False
if res_list[i][0] == res_list[j][0] and res_list[i][1] != res_list[j][1]:
return False
return True