代码随想录算法训练营第七天| 454. 四数相加 II、383. 赎金信、15. 三数之和、18. 四数之和

454. 四数相加 II：

---

代码思路

哈希表的很多思路都是记录出现的次数，这题也不例外。
该题的思路极限只能降到 O(2*n^2) 。原本是n的四次方。
因为将四个数组分为两组，然后用两个哈希表分别存储这两组的结果，key是相加的结果（为了确保key相同，第二个哈希表的key取加和的相反数），value是出现这个结果的次数。最后哈希对应key的value相乘（排列组合中的乘法公式）。

class Solution:
    def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:

        from collections import defaultdict
        hashset1 = defaultdict(int)
        for i in nums1:
            for j in nums2:
                hashset1[i+j] += 1

        hashset2 = defaultdict(int)
        for i in nums3:
            for j in nums4:
                hashset2[-(i+j)] += 1
        res = 0
        for k, v in hashset1.items():
            if hashset1[k] != 0 and hashset2[k] != 0:
                tmp = hashset1[k] * hashset2[k]
                res += tmp

        return res

383. 赎金信

---

代码思路

也是传统哈希表的思路，记录次数。
这次记录的次数是magazine中字符出现的次数。
题目的意思其实就是ransomNote的字符，magazine里面都要有。
所以直接记录magazine中每个字母的出现次数。当ransomNote出现新字符，哈希表中该字符的次数肯定小于零。

class Solution:
    def canConstruct(self, ransomNote: str, magazine: str) -> bool:
        from collections import defaultdict
        hashmap = defaultdict(int)
        for i in magazine:
            hashmap[i] += 1
        for i in ransomNote:
            hashmap[i] -= 1
            if hashmap[i] < 0:
                return False
        return True

15. 三数之和

---

代码思路

三数之和和四数之和的思想都是降低一个指数值的复杂度。比如三数之和就是n三次方变成n二次方。
双指针玩法就是做这种事情的。
首先数组加和的题的套路都是先排序加双指针。
主要是双指针的话要固定遍历一个数值，每次遍历当中动两个数值，不然就是三指针了。

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        nums.sort()        
        res = []
        for k in range(len(nums) - 2):
            if k-1 >= 0 and nums[k] == nums[k-1]:
                continue
            i = k + 1
            j = len(nums) - 1
            while i < j:
                if nums[k] + nums[i] + nums[j] > 0:
                    j -= 1
                elif nums[k] + nums[i] + nums[j] < 0:
                    i += 1
                else:
                    if nums[i] == nums[i - 1] and i - 1 >= k + 1:
                        j -= 1
                        i += 1
                        continue
                    res.append([nums[k],nums[i],nums[j]])
                    i += 1
                    j -= 1
        return res

18. 四数之和

代码思路

也是排序加双指针。
这题是固定遍历两个数值，动两个数值。

class Solution:
    def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
        nums.sort()
        res = []

        for i in range(len(nums) - 3):
            if i - 1 >= 0 and nums[i] == nums[i - 1]:
                continue
            for j in range(i+1, len(nums) - 2):
                if j - 1 >= i+1 and nums[j] == nums[j - 1]:
                    continue
                left = j + 1
                right = len(nums) - 1
                while left < right:
                    if nums[i] + nums[j] + nums[left] + nums[right] < target:
                        left += 1
                        continue
                    elif nums[i] + nums[j] + nums[left] + nums[right] > target:
                        right -= 1
                        continue
                    else:
                        if left - 1 >= j + 1 and nums[left] == nums[left-1]:
                            left += 1
                            right -= 1                           
                            continue
                        res.append([nums[i], nums[j], nums[left], nums[right]])
                        left += 1
                        right -= 1
        return res