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

LeetCode 454-四数相加II

题目:
给定四个包含整数的数组列表 A , B , C , D ,计算有多少个元组 (i, j, k, l) ，使得 A[i] + B[j] + C[k] + D[l] = 0。

为了使问题简单化，所有的 A, B, C, D 具有相同的长度 N，且 0 ≤ N ≤ 500 。所有整数的范围在 -2^28 到 2^28 - 1 之间，最终结果不会超过 2^31 - 1 。

题目链接：454-四数相加II

思路：
把四个数组的和变成每两个数组和的相加。遍历数组a和b并相加，统计数字和出现的次数。后遍历c和d数组并相加在字典中找出符合-(c+d)的数字，统计该数字出现的次数。

class Solution:
    def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:
        hashmap = dict()
        for n1 in nums1:
            for n2 in nums2:
                if n1 + n2 in hashmap:
                    hashmap[n1 + n2] += 1
                else:
                    hashmap[n1 + n2] = 1

        count = 0

        for n3 in nums3:
            for n4 in nums4:
                key = - n3 -n4
                if key in hashmap:
                    count += hashmap[key]
        return count

LeetCode 383-赎金信

题目：
给定一个赎金信 (ransom) 字符串和一个杂志(magazine)字符串，判断第一个字符串 ransom 能不能由第二个字符串 magazines 里面的字符构成。如果可以构成，返回 true ；否则返回 false。

题目链接：383-赎金信

思路：
本题可以采用数组或者字典实现，可以使用数组是因为字符串仅由小写字母组成，因此可以采用26位的数组来储存出现过的字符。
总体思路就是，先记录在magazine出现的字符次数，再与ransom字符串中出现的字母进行比较，若ransom出现的字符在magazine没出现或者出现的次数多于在magazine出现的次数则返回false，否则返回true。

class Solution:
    def canConstruct(self, ransomNote: str, magazine: str) -> bool:
        # arr = [0] * 26 #ransomNote 和 magazine 由小写英文字母组成

        # for str1 in magazine:
        #     arr[ord(str1) - ord('a')] += 1
        
        # for str2 in ransomNote:
        #     if arr[ord(str2) - ord('a')] == 0:
        #         return False
        #     else:
        #         arr[ord(str2) - ord('a')] -= 1

        # return True
        hashmap = dict()
        for str1 in magazine:
            if str1 in hashmap:
                hashmap[str1] += 1
            else:
                hashmap[str1] = 1

        for str2 in ransomNote:
            if str2 not in hashmap:
                return False
            else:
                hashmap[str2] -= 1 
            if hashmap[str2] < 0:
                return False
        
        return True

LeetCode 15-三数之和

题目：
给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有满足条件且不重复的三元组。

注意： 答案中不可以包含重复的三元组。

题目链接：15-三数之和

思路：
首先对数组进行排序。采用双指针的解法，以i为起点进行遍历，双指针的起点设置为i+1，终点为数组的末端，若i指针所指结点大于零则将i右移，否则将三数相加，和为0则存起来，和大于0则左移右指针，否则右移左指针。遇到相同的数字需要去重，即遇到相同的数字，指针继续移动。

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        ans = []
        n = len(nums)
        nums.sort() #先保证数组是有序的

        for i in range(n):
            left = i + 1
            right = n - 1
            if nums[i] > 0:
                break
            if i >= 1 and nums[i] == nums[i - 1]:
                continue
            while left < right:
                Sum = nums[i] + nums[left] + nums[right]
                if Sum > 0:
                    right -= 1
                elif Sum < 0:
                    left += 1
                else:
                    ans.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
                    left += 1
                    right -= 1
        return ans

LeetCode 18-四数之和

题目：
题意：给定一个包含 n 个整数的数组 nums 和一个目标值 target，判断 nums 中是否存在四个元素 a，b，c 和 d ，使得 a + b + c + d 的值与 target 相等？找出所有满足条件且不重复的四元组。

注意：
答案中不可以包含重复的四元组。

题目链接：18-四数之和

思路：
首先将数组排序。类似于上题我们需要构建两个指针进行循环，在循环中继续运用双指针将四数相加。去重：若遍历到的数字与上一个数字相同，则继续遍历。

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

        for i in range(n):
            if i > 0 and nums[i] == nums[i - 1]:
                continue
            for k in range(i+1, n):
                if k > i + 1 and nums[k] == nums[k - 1]:
                    continue
                p = k + 1
                q = n - 1

                while p < q:
                    if nums[i] + nums[k] + nums[p] + nums[q] > target:
                        q -= 1
                    elif nums[i] + nums[k] + nums[p] + nums[q] < target:
                        p += 1
                    else:
                        ans.append([nums[i], nums[k], nums[p], nums[q]])

                        while p < q and nums[p] == nums[p + 1]: p += 1
                        while p < q and nums[q] == nums[q - 1]: q -= 1
                        p += 1
                        q -= 1
        return ans

总结：
今天学习到了去重的思想，对哈希表的使用更加熟练了一些。