题目
昨天做力扣新手村刷到的题目:
给你两个字符串:ransomNote 和 magazine ,判断 ransomNote 能不能由 magazine 里面的字符构成。
如果可以,返回 true ;否则返回 false 。
magazine 中的每个字符只能在 ransomNote 中使用一次。
示例 1:
输入:ransomNote = "aa", magazine = "ab"
输出:false
示例 2:
输入:ransomNote = "aa", magazine = "aab"
输出:true
我的垃圾代码
这是我一开始的解答:
class Solution(object):
def canConstruct(self, ransomNote, magazine):
mag=[]
ran=[]
for i in magazine:
mag.append(i)
for i in ransomNote:
ran.append(i)
ran1=tuple(ran)
for i in ran1:
for n in mag:
if i == n:
mag.remove(i)
ran.remove(i)
break
if ran==[]:
return True
return False
就是单纯的暴力解法,创建了两个列表,依次取出 ransomNote中的元素,如果magazine中有相同的元素,就同时删除。最后如果ransomNote不再有剩余元素,那么就返回True,否则为False。
这个结果嘛,可想而知:
然后进行了一些改进:
class Solution(object):
def canConstruct(self, ransomNote, magazine):
mag=[]
ran=[]
for i in magazine:
mag.append(i)
for i in ransomNote:
ran.append(i)
try:
for i in ran:
mag.remove(i)
return True
except:
return False
用了try-except,只遍历ransomNote,然后删除magazine中的元素。如果循环正常结束,那说明ransomNote中的元素在magazine中都存在,自然是返回True,否则就是False。
跑一下看看:
有进步,但不多(赏自己一个大嘴巴子)
大佬的解法
走投无路之下,看了一下大佬的解答:
class Solution(object):
def canConstruct(self, ransomNote, magazine):
ransomNote_set = set(ransomNote)
for i in ransomNote_set:
if magazine.count(i) < ransomNote.count(i):
return False
return True
用了两个函数set()和.count()。大佬先用set()函数对ransomNote进行了去重。然后对集合中的元素进行了遍历(才知道Python3可以遍历集合)。依次对ransomNote和magazine中元素的个数进行统计及比较。如果某元素在magazine中的个数比ransomNote中该元素的个数少,那么即为False,否则为True。
效果如下:
(我写的果真就是依托答辩)
总结一下
以前写程序很少会想到使用集合,但对于这种题来讲,集合+元素数量判断确实是一个极好的办法,以后再写程序时也可以尝试更多的使用集合,以及,集合可以遍历(再来一个大嘴巴子)。