想躺平又躺不平的day05

算法随想录之哈希表-进阶版


四数相加 II

四个整数数组 nums1、nums2、nums3 和 nums4,数组长度都是 n,计算有多少个元组 (i, j, k, l) 能满足:

  • 0 <= i, j, k, l < n
  • nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0

提示:这道题只需要返回有多少组元组即可,那么也就是说可以将四数相加转换为两数相加问题。

  遍历其中两个数组得到两两之间的和存进map哈希结构,key为和,value为key对应出现的次数。接着,同样遍历另外两个数组,并求两两之间的和s判断(0-s)是否在哈希表中并返回对应的value值为当前符合条件的元组数,累加得到最终的元组数。

class Solution(object):
    def fourSumCount(self, nums1, nums2, nums3, nums4):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :type nums3: List[int]
        :type nums4: List[int]
        :rtype: int
        """
        hash_table = {}
        for n1 in nums1:  # 前两个数组和
            for n2 in nums2:
                hash_table[n1+n2] = hash_table.get(n1+n2,0) + 1
        count = 0
        for n3 in nums3: # 后两个数组和
            for n4 in nums4:
                s = n3 + n4
                k = -s   # 判断条件(0-s)
                if k in hash_table:
                    count += hash_table[k]
        return count

赎金信

两个字符串:ransomNote 和 magazine ,判断 ransomNote 能不能由 magazine里面的字符构成。如果可以,返回 true ;否则返回 false 。magazine 中的每个字符只能在 ransomNote 中使用一次。

  这道题与【字母异位词】解题思想一致:以字母的ASCII码特性将不同字母与‘a’的ASCII码差值得到索引值,存入哈希表数组对应的索引下。要判断ransomNote字符串是否由Magazine字符串构成,即判断ransomNote中字母出现的次数是否小于等于Magazine字符串中字母出现的次数。

class Solution(object):
    def canConstruct(self, ransomNote, magazine):
        """
        :type ransomNote: str
        :type magazine: str
        :rtype: bool
        """
        hash_table = [0] * 26
        for i in range(len(ransomNote)):  # 统计ransomNote中字母出现的次数
            hash_table[ord(ransomNote[i])-ord('a')] += 1
        for j in range(len(magazine)):    # 在ransomNote基础上减去magazine字母出现的次数
            hash_table[ord(magazine[j])-ord('a')] -= 1
        for h in hash_table:  # 哈希表中元素需全都小于等于0
            if h > 0:  
                return False
        return True

三数之和

给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请返回所有和为 0 且不重复的三元组。
**注意:**答案中不可以包含重复的三元组。

  拿到这道题下意识地就想着用哈希表解决,但是发现,首先同一索引的元素不能取多次,然后得到的三个数字的组合要去重。我直接一整个呆住。经过一番折腾去看了题解(实在是尽力了)。。。。。。
  这道题的解题思路是先将数组排序,然后第一个元素取值为整个数组。索引表示为i,剩下的两个元素只能取第一元素后面的元素即[i+1,len(nums)-1]。利用双指针,左指针指向i+1,右指针指向len(nums)-1。这样,就能保证同一索引的值不会取多次。
  然后,由于一开始我们已经对数组进行排序,此时将三个索引对应的值相加判断是否等于target,小于target其他两个值不动,将左指针往右加一;大于target则将右指针往左减一。如果等于target则记录下来。
在这里插入图片描述
  划重点:去重的逻辑主要考虑三个数的去重,保证三个元素对应的位置不会再次取到相同的元素(比如说第一个位置取了某个值n,双指针根据遍历条件得到符合条件的元素[a,b],那么第一个位置的数往后遍历则不能再取n了,会得到重复的结果),但是三个元素的值可以相同(来源于不同索引的相同值)。

  • 在第一位置的元素去重的条件为nums[i-1] == nums[i],即不能取与之前取值相同的元素(因为已经排过序,所以只用比上一个索引对应的值而不用与之前所有的值进行对比)。同时,由于要比较上一索引,所以还需加一个条件i>0。
  • 同理,双指针对应的两个元素也不能重复取已经取过的元素,所以在当两个指针当前值和下一个值相等时,将两个指针分别往右和往左移动一个位置,直到不相等为止。此时的左右指针指向的为重复元素的最后一个。这样,达到了对第二、第三个元素去重的目的。
class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        nums.sort()
        result = []
        for i in range(len(nums)):
            if nums[i] > 0:  # 剪枝(如果第一个数为正数,那么加上后面两个数之和不可能等于0)
                return result
            if i > 0 and nums[i-1] == nums[i]:  # 第一个元素的去重
                continue
            left = i + 1
            right = len(nums) - 1
            while left < right:
                s = nums[i] + nums[left] + nums[right]
                if s > 0:
                    right -= 1
                elif s < 0:
                    left += 1
                else:
                    result.append([nums[i],nums[left],nums[right]])
                    while left < right and nums[left] == nums[left+1]: # 第二个元素的去重
                        left += 1
                    while left < right and nums[right] == nums[right-1]: # 第三个元素的去重
                        right -= 1
                    right -= 1
                    left += 1
        return result

四数之和

给你一个由 n 个整数组成的数组 nums ,和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组[nums[a], nums[b], nums[c], nums[d]] (若两个四元组元素一一对应,则认为两个四元组重复):

  • 0 <= a, b, c, d < n
  • a、b、c 和 d 互不相同
  • nums[a] + nums[b] + nums[c] + nums[d] == target 可以按 任意顺序 返回答案 。

  根据上述三数之和的思想,四数之和即在三数之和的基础上再套一层循环:第一个数字遍历整个数组,第二个数字遍历从第一个数字取值的后一位到数组最后一位,然后剩下两个数字由左、右双指针进行判断。
  这道题还需要注意的地方就是target并不是大于等于0的,因此剪枝的时候不能直接像上述三数之和那样剪枝,而是需要加一个target>=0的约束条件进行剪枝。

class Solution(object):
    def fourSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        nums.sort()
        result = []
        n = len(nums)
        for i in range(n):
            if nums[i] > target and target >= 0: # 剪枝(如果第一个数大于target,那么加上后面两个数之和(此时一定为正数)不可能等于target(需保证target>=0))
                return result  #此时剪枝后就无需再往后遍历了,直接返回结果
            if i > 0 and nums[i] == nums[i-1]:  # 第一个元素的去重
                continue
            for j in range(i+1,n):
                if nums[i]+nums[j] > target and target >= 0: # 剪枝(与上述思想一致,但是比较对象编程前两个数之和)
                    break  #此时剪枝后对于j就无需再往后遍历了,但是i仍然需要往后遍历,所以break而不是return result
                if j > i+1 and nums[j] == nums[j-1]:
                    continue
                left = j+1
                right = len(nums)-1
                while left < right:
                    s = nums[i] + nums[j] + nums[left] + nums[right]
                    if s == target:
                        result.append([nums[i], nums[j], nums[left], nums[right]])
                        while left < right and nums[left] == nums[left+1]:  # 第二个元素的去重
                            left += 1
                        while left < right and nums[right] == nums[right-1]:  # 第三个元素的去重
                            right -= 1
                        left += 1
                        right -= 1
                    elif s < target:
                        left += 1
                    else:
                        right -= 1
        return result

  我自己在写这个代码的时候,就发现老是有一部分样例通不过,但是又对比代码感觉没什么差别。后来一看,才发现在剪枝处理的时候,最外层元素的剪枝可以直接返回result,但是里层的剪枝却不能直接返回result,而是应该用break。因为,对于里层的遍历进行剪枝了,但是外层的遍历还要往后遍历。

  • 20
    点赞
  • 26
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值