leetcode454. 四数相加 II
本质上是对于lc1 两数之和的一次延申问,重点在于对四个数组进行分组,使用字典记录各个各个组合出现的次数返回累加值。
//使用字典来完成对于前两个数字的记载,然后对于后两个数字进行加和,在前方字典中检索出现次数并累加。
class Solution:
def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:
hashmap = {}
for i in nums1:
for j in nums2:
# hashmap[i+j] +=1
if i+j in hashmap:
hashmap[i+j] += 1
else:
hashmap[i+j] = 1
cnt = 0
for i in nums3:
for j in nums4:
neg = -i -j
if neg in hashmap:
cnt += hashmap[neg]
return cnt
// 优化字典初始化的方式,使用get()函数简化判断。
for n1 in nums1:
for n2 in nums2:
hashmap[n1+n2] = hashmap.get(n1+n2, 0) + 1
leetcode 383 赎金信
使用字典记录较长的字符串内容,而后取出较短字符串中的内容进行对比。
class Solution:
def canConstruct(self, ransomNote: str, magazine: str) -> bool:
cnt = len(ransomNote)
ori = {}
for word in magazine:
ori[word] = ori.get(word,0) + 1
for word in ransomNote:
if word in ori:
if ori[word] == 0:
return False
else:
ori[word] -= 1
cnt -= 1
if cnt ==0:
return True
else:
return False
//使用Counter同样可以完成,快捷,推荐多去了解
from collections import Counter
class Solution:
def canConstruct(self, ransomNote: str, magazine: str) -> bool:
return not Counter(ransomNote) - Counter(magazine)
leetcode 15 三数之和 梦碎之处
与前几道题目不同,我们如果使用哈希解法,我们可以使用两层for循环实现a和b两个数字的确认,使用哈希法确认0-(a+b)是否在数组中出现过,但是这个思路有一个问题,如何在取得全部三元组后进行去重,十分复杂。
所以更换思路,使用双指针法实现目标。拿这个nums数组来举例,首先将数组排序,然后有一层for循环,i从下标0的地方开始,同时定一个下标left 定义在i+1的位置上,定义下标right 在数组结尾的位置上。
依然还是在数组中找到 abc 使得a + b +c =0,我们这里相当于 a = nums[i],b = nums[left],c = nums[right]。
接下来如何移动left 和right呢, 如果nums[i] + nums[left] + nums[right] > 0 就说明 此时三数之和大了,因为数组是排序后了,所以right下标就应该向左移动,这样才能让三数之和小一些。
如果 nums[i] + nums[left] + nums[right] < 0 说明 此时 三数之和小了,left 就向右移动,才能让三数之和大一些,直到left与right相遇为止。
逻辑并不复杂,那么如何实现去重呢。
从i开始,假设nums[i]的数值为a,如果之前遇到过a,那么在向右移动时再次遇到a时所得到的三元组必定会重复,需要直接跳过这一数值,那么如何设置判断条件。以下是两种判断形式,分别是判断和后一个数字相同还是和前一个数字相同,举个例子{-1,-1,2},如果和后一个判断,无解,但是和前一个判断,就没有问题,可想而知,和前一位数做判断,能保证这个数值作用的有效性。
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
而对于后两个数值的去重,假设为b和c,只有当三元组之和为0时才需要,不为零时指针依旧会以动,即使重复也无伤大雅。主要问题是,去重过程中记得保证判断条件right>left,减少多余的运算以及索引溢出数组的风险。
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
result = []
nums.sort()
for i in range(len(nums)):
if nums[i]>0:
break
if i > 0 and nums[i] == nums[i - 1]:
continue
left = i + 1
right = len(nums) - 1
while left < right:
sum_ = nums[i]+nums[left]+nums[right]
if sum_>0:
right -= 1
elif sum_<0:
left +=1
else:
result.append([nums[i],nums[left],nums[right]])
while right > left and nums[right] == nums[right-1]:
right -= 1
while right > left and nums[left] == nums[left+1]:
left += 1
left += 1
right -= 1
return result
leetcode 18 四数之和
本质上和三数之和差异不大,更多的是多出一位j来确认第二个数值的信息,更多的重点落在了去重(和三数之和方式一致)以及区间的判定上,还有计数最终结果是target不是0。同样可以使用 部分剪枝操作完成算法的速度优化。
class Solution:
def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
result = []
nums.sort()
n = len(nums)
for i in range(n-3):
if nums[i] > target and nums[i] > 0 and target > 0:
break
if i > 0 and nums[i] == nums[i-1]:
continue
for j in range(i+1,n-2):
if j>i+1 and nums[j] == nums[j-1]:
continue
left = j+1
right = n-1
while left < right:
sum_ = nums[i]+nums[j]+nums[left]+nums[right]
if sum_ < target:
left += 1
elif sum_ > target:
right -= 1
else:
result.append([nums[i],nums[j],nums[left],nums[right]])
while left<right and nums[right]==nums[right-1]:
right -= 1
while right> left and nums[left] == nums[left+1]:
left += 1
right -= 1
left += 1
return result