文章目录
- 题目描述
- 思路分析
- 完整代码
题目描述
你有一个单词列表 words 和一个模式 pattern,你想知道 words 中的哪些单词与模式匹配。
如果存在字母的排列 p ,使得将模式中的每个字母 x 替换为 p(x) 之后,我们就得到了所需的单词,那么单词与模式是匹配的。
(回想一下,字母的排列是从字母到字母的双射:每个字母映射到另一个字母,没有两个字母映射到同一个字母。)
返回 words 中与给定模式匹配的单词列表。
你可以按任何顺序返回答案。
示例:
输入:words = [“abc”,“deq”,“mee”,“aqq”,“dkd”,“ccc”], pattern = “abb”
输出:[“mee”,“aqq”]
解释:
“mee” 与模式匹配,因为存在排列 {a -> m, b -> e, …}。
“ccc” 与模式不匹配,因为 {a -> c, b -> c, …} 不是排列。
因为 a 和 b 映射到同一个字母。
思路分析
这道题我看到之后直接想用俩哈希表或者单个哈希表映射来做,后来发现评论区有诡异大佬,用zip做的,于是我来效仿一下。
假设有:words = [“mee”,], pattern = “abb”
则
list(zip(words,pattern)后有:
[(‘m’, ‘a’), (‘e’, ‘b’), (‘e’, ‘b’)]
去重之后
set(zip(words,pattern)后有:
{(‘m’, ‘a’), (‘e’, ‘b’)}
看到这应该都懂了,只要在对应映射去重之后长度 = 单词去重长度 = 模式去重长度 那么这俩就是相同的模式。
完整代码
class Solution:
def findAndReplacePattern(self, words: List[str], pattern: str) -> List[str]:
return [i for i in words if len(set(i)) == len(set(pattern)) == len(set(zip(i,pattern)))]
# res = []
# for i in words:
# if len(set(i)) == len(set(pattern)) == len(set(zip(i,pattern))):
# res.append(i)
# print(res)
# return res