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

454. 四数相加 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，遍历数组 1 和 2，记录两数相加出现的次数。再遍历数组 3 和 4，定义 count 记录 [0-(v3+v4)] 出现的次数。count 就是满足条件的个数
实现：

func fourSumCount(nums1 []int, nums2 []int, nums3 []int, nums4 []int) int {
    m := make(map[int]int)
    count := 0
    for _, v1 := range nums1{
        for _, v2 := range nums2{
            m[v1+v2]++
        }
    }

    for _, v3 := range nums3{
        for _, v4 := range nums4{
            count += m[-v3-v4]
        }
    }
    return count
}

383. 赎金信

类型： #哈希表
难度： #简单
题目：给你两个字符串：ransomNote 和 magazine ，判断 ransomNote 能不能由 magazine 里面的字符构成。如果可以，返回 true ；否则返回 false 。magazine 中的每个字符只能在 ransomNote 中使用一次。
思路：本质上是单向判断字符串 a 能否组成字符串 b。用一个 26 位数组，遍历记录 magazine 中字母出现次数。再遍历 ransomNote 将字母出现次数减去，如果 <0 说明有不一样的字母出现
实现：

func canConstruct(ransomNote string, magazine string) bool {
    record := make([]int, 26)
    for _, v := range magazine{
        record[v-'a']++
    }

    for _, v := range ransomNote{
        record[v-'a']--
        if record[v-'a'] < 0{
            return false
        }
    }

    return true
}

15. 三数之和

类型： #哈希表 #双指针
难度： #中等
题目：给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时同时还满足 nums[i] + nums[j] + nums[k] == 0 。请
你返回所有和为 0 且不重复的三元组。注意：答案中不可以包含重复的三元组。
思路：用双指针。第一个指针 i 指向数组开头，左指针 l 指向 i+1，右指针 r 指向 数组长度 -1，如图让指针不断去移动。遍历过程中要对三元组去重，去重 a 主要是与上一个 i 相比是否重复，是的话跳过；去重 b 和 c 是判断是否相等，相等的话 l++ 或者 r–

实现：

func threeSum(nums []int) [][]int {
    sort.Ints(nums)
    res := [][]int{}

    for i:=0; i<len(nums)-2; i++{
        n1 := nums[i]
        if n1 > 0{
            break
        }
        if i>0 && n1 == nums[i-1]{
            continue
        }
        l, r := i+1, len(nums)-1
        for l < r{
            n2, n3 := nums[l], nums[r]
            if n1+n2+n3==0{
                res = append(res, []int{n1, n2, n3})
                for l<r && nums[l] == n2{
                    l++
                }
                for l<r && nums[r] == n3{
                    r--
                }
            }else if n1+n2+n3 <0{
                l++
            }else{
                r--
            }

        }
    }
    return res
}

18. 四数之和

类型： #哈希表 #双指针
难度： #中等
题目：给你一个由 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

思路：与 15. 三数之和 类似，多一个 j 循环，同时要记得找到一个之后，l 和 r 要收缩
实现：

func fourSum(nums []int, target int) [][]int {
    if len(nums) < 4{
        return nil
    }
    sort.Ints(nums)
    res := [][]int{}

    for i:=0; i<len(nums)-3; i++{
        n1 := nums[i]
        if i>0 && n1==nums[i-1]{
            continue
        }
        for j:=i+1;j<len(nums)-2;j++{
            n2 := nums[j]
            if j>i+1 && n2==nums[j-1]{
                continue
            }
            l, r := j+1, len(nums)-1
            for l < r{
                n3, n4 := nums[l], nums[r]
                sum := n1 + n2 + n3 + n4
                if sum < target{
                    l++
                }else if sum > target{
                    r--
                }else{
                    res = append(res, []int{n1, n2, n3, n4})
                    for l<r && n3==nums[l+1]{
                        l++
                    }
                    for l<r && n4==nums[r-1]{
                        r--
                    }
                    r--
                    l++
                }
            }
        }
    }
    return res
}