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

semirc9

于 2024-06-27 21:55:10 发布

阅读量1.2k

点赞数 27

分类专栏：代码随想录算法训练营文章标签：算法 c语言数据结构哈希表

本文链接：https://blog.csdn.net/semirc9/article/details/140025384

版权

代码随想录算法训练营专栏收录该内容

8 篇文章 0 订阅

订阅专栏

文章目录

454.四数相加II
- 思路
383.赎金信
- 解法--哈希表
15.三数之和
- 解法一、哈希表+去重
- 解法二、排序+双指针
18.四数之和
- 排序+双指针

454.四数相加II

题目链接：454. 四数相加 II - 力扣（LeetCode）
题目描述：给你四个整数数组 nums1、nums2、nums3 和 nums4 ，数组长度都是 n ，请你计算有多少个元组 (i, j, k, l) 能满足：

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

示例 1：

输入：nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2]
输出：2
解释：
两个元组如下：
1. (0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0
2. (1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0

示例 2：

输入：nums1 = [0], nums2 = [0], nums3 = [0], nums4 = [0]
输出：1

提示：

n == nums1.length
n == nums2.length
n == nums3.length
n == nums4.length
1 <= n <= 200
-228 <= nums1[i], nums2[i], nums3[i], nums4[i] <= 228

思路

分解为两个两数之和的问题

首先定义一个unordered_map，key放a和b两数之和，value 放a和b两数之和出现的次数。
遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中。
定义int变量count，用来统计 a+b+c+d = 0 出现的次数。
再遍历大C和大D数组，找到如果 0-(c+d) 在map中出现过的话，就用count把map中key对应的value也就是出现次数统计出来。
最后返回统计值 count 就可以了

struct hashTable {
    int key;
    int val;
    UT_hash_handle hh;
};
int fourSumCount(int* A, int ASize, int* B, int BSize, int* C, int CSize, int* D, int DSize) {
    struct hashTable* hashtable = NULL;
    for (int i = 0; i < ASize; ++i) 
    {
        for (int j = 0; j < BSize; ++j) 
        {
            int ikey = A[i] + B[j];
            struct hashTable* hash;
            HASH_FIND_INT(hashtable, &ikey, hash);
            if (hash == NULL) {
                struct hashTable* hash = malloc(sizeof(struct hashTable));
                hash->key = ikey;
                hash->val = 1;
                HASH_ADD_INT(hashtable, key, hash);
            } 
            else 
            {
                hash->val++;
            }
        }
    }
    int count = 0;
    for (int i = 0; i < CSize; ++i) 
    {
        for (int j = 0; j < DSize; ++j) 
        {
            int ikey = -C[i] - D[j];
            struct hashTable* hash;
            HASH_FIND_INT(hashtable, &ikey, hash);
            if (hash != NULL) 
            {
                count += hash->val;
            }
        }
    }
    return count;
}

383.赎金信

题目链接：383. 赎金信 - 力扣（LeetCode）

题目描述：给你两个字符串：ransomNote 和 magazine ，判断 ransomNote 能不能由 magazine 里面的字符构成。

如果可以，返回 true ；否则返回 false 。

magazine 中的每个字符只能在 ransomNote 中使用一次。

示例 1：

输入：ransomNote = "a", magazine = "b"
输出：false

示例 2：

输入：ransomNote = "aa", magazine = "ab"
输出：false

示例 3：

输入：ransomNote = "aa", magazine = "aab"
输出：true

解法–哈希表

bool canConstruct(char* ransomNote, char* magazine) {
    int rlen = strlen(ransomNote);
    int mlen = strlen(magazine);
    if (rlen > mlen) {
        return false;
    }
    int ans[26] = {0};
    for (int i = 0; i < rlen; i++) {
        ans[ransomNote[i] - 'a']++;
        ans[magazine[i] - 'a']--;
    }
    for (int i = rlen; i < mlen; i++) {
        ans[magazine[i] - 'a']--;
    }
    for (int i = 0; i <26; i++) {
        if (ans[i] > 0) {
            return false;
        }
    }
    return true;
}

15.三数之和

题目链接：15. 三数之和 - 力扣（LeetCode）
题目描述：给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

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

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
解释：
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意，输出的顺序和三元组的顺序并不重要。

示例 2：

输入：nums = [0,1,1]
输出：[]
解释：唯一可能的三元组和不为 0 。

示例 3：

输入：nums = [0,0,0]
输出：[[0,0,0]]
解释：唯一可能的三元组和为 0 。

提示：

3 <= nums.length <= 3000
-105 <= nums[i] <= 105

解法一、哈希表+去重

本题用哈希表方法会比较复杂，代码之后再更新

解法二、排序+双指针

首先将数组排序，然后有一层for循环，i从下标0的地方开始，同时定义左指针left 定义在i+1的位置上，定义右指针right 在数组结尾的位置上。

如何移动left和right？

若nums[i] + nums[left] + nums[right] > 0 说明此时三数之和大了，因为数组是排序后了，所以right应该向左移动，减小三数之和

若nums[i] + nums[left] + nums[right] < 0 说明此时三数之和小了，left就向右移动，才能让三数之和大一些，直到left与right相遇为止。

去重问题

题目说的是答案中不可以包含重复的三元组，因此每个三元组中的元素是可以重复的。
这里主要考虑nums[i]、nums[left]、nums[right]的去重。

对于nums[i],有两种方法判断重复，即判断nums[i]与nums[i + 1]是否相同或者判断 nums[i]与nums[i-1]是否相同，二者看似一样，实则不同。

如果这样写：

if (nums[i] == nums[i + 1]) { // 去重操作
    continue;
}

就把三元组中出现重复元素的情况直接pass掉了。例如{-2, -2 ,4} 这组数据，当遍历到第一个-2 的时候，判断下一个也是-2，那这组数据就pass了。

因此对nums[i]的去重应这样写：

if (i > 0 && nums[i] == nums[i - 1]) {
    continue;
}

这么写就是当前使用 nums[i]，我们判断前一位是不是一样的元素，在看 {-2, -2 ,4} 这组数据，当遍历到第一个 -2 的时候，只要前一位没有-2，那么 {-2, -2 ,4} 这组数据一样可以收录到结果集里。

左右指针的去重

    while(left<right&&nums[right]==nums[right-1]){
        right--;
    }
    while(left<right&&nums[left]==nums[left+1]){
        left++;
    }

完整代码如下

int cmp(const void *a,const void *b){
    return *(int*)a - *(int*)b;
}
int** threeSum(int* nums, int numsSize, int* returnSize, int** returnColumnSizes) {
    qsort(nums,numsSize,sizeof(int),cmp);
    int **ans=(int**)malloc(sizeof(int*)*numsSize);
    int cnt=0;
    if(numsSize<3){
        *returnSize=0;
        return  ans;
    }
    for(int i=0;i<numsSize-2;i++){
        if(nums[i]>0){ //若当前i指向元素>0，则代表left和right以及i的和大于0。直接break
            break;
        }
        if(i>0&&nums[i]==nums[i-1]){ //去重
            continue;
        }
        int left=i+1;
        int right=numsSize-1;
        while(left<right){
            int sum=nums[i]+nums[left]+nums[right];
            if(sum>0){
                right--;
            }else if(sum<0){
                left++;
            }else{
                ans[cnt]=(int*)malloc(sizeof(int)*3);
                ans[cnt][0]=nums[i];
                ans[cnt][1]=nums[left];
                ans[cnt][2]=nums[right];
                cnt++;
                //去重
                while(left<right&&nums[right]==nums[right-1]){
                    right--;
                }
                while(left<right&&nums[left]==nums[left+1]){
                    left++;
                }
                //更新左右指针
                left++;
                right--;
            }
        }
    }
    //设定返回数组大小
    *returnSize=cnt;
    *returnColumnSizes=(int*)malloc(sizeof(int)*cnt);
    for(int i=0;i<cnt;i++){
        (*returnColumnSizes)[i]=3;
    }
    return ans;
}

18.四数之和

题目链接：18. 四数之和 - 力扣（LeetCode）
题目描述：给你一个由 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

你可以按 任意顺序 返回答案。

示例 1：

输入：nums = [1,0,-1,0,-2,2], target = 0
输出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

示例 2：

输入：nums = [2,2,2,2,2], target = 8
输出：[[2,2,2,2]]

提示：

1 <= nums.length <= 200
-109 <= nums[i] <= 109
-109 <= target <= 109

排序+双指针

本题和15. 三数之和 - 力扣（LeetCode）是一个思路，就是在三数之和的基础上再加一层for循环

四数之和的双指针解法是两层for循环nums[k] + nums[i]为确定值，依然是循环内有left和right下标作为双指针，找出nums[k] + nums[i] + nums[left] + nums[right] == target的情况

int cmp(const void *a,const void *b){
    return *(int*)a - *(int*)b;
 }
int** fourSum(int* nums, int numsSize, int target, int* returnSize, int** returnColumnSizes) {
    qsort(nums,numsSize,sizeof(int),cmp);
    int **ans=(int**)malloc(sizeof(int*)*1001);
    if(numsSize<4){
        *returnSize=0;
        return ans;
    }
    long int cnt=0;
    for(int i=0;i<numsSize-3;i++){
        //剪枝优化
        if(nums[i]>target&&target>=0){
            break;
        }
        //对nums[i]去重
        if(i>0&&nums[i]==nums[i-1]){
            continue;
        }
        for(int j=i+1;j<numsSize-2;j++){
             // 2级剪枝处理
            if (nums[j] + nums[i] > target && nums[j] + nums[i] >= 0) {
                break;
            }
            if(j>i+1&&nums[j]==nums[j-1]){
                continue;
            }
            int left=j+1;
            int right=numsSize-1;
            while(right>left){
                //注意整数溢出问题
               long int sum=(long)nums[i]+nums[j]+nums[left]+nums[right];
                if(sum>target){
                    right--;
                }else if(sum<target){
                    left++;
                }else{
                    int *arr=(int*)malloc(sizeof( int)*4);
                    arr[0]=nums[i],arr[1]=nums[j],arr[2]=nums[left],arr[3]=nums[right];
                    ans[cnt++]=arr;
                    // 对nums[left]和nums[right]去重
                    while (right > left && nums[right] == nums[right - 1]){
                        right--;
                    }
                    while (right > left && nums[left] == nums[left + 1]) {
                        left++;
                    }
                    // 找到答案时，双指针同时收缩
                    right--;
                    left++;
                }
            }
        }
    }
    *returnSize=cnt;
    *returnColumnSizes=(int*)malloc(sizeof(int)*cnt);
    for(int i=0;i<cnt;i++){
        (*returnColumnSizes)[i]=4;
    }
    return ans;
}

semirc9

关注

27
点赞
踩
27

收藏

觉得还不错? 一键收藏
0
评论
代码随想录算法训练营第六天|454.四数相加II、383.赎金信、15.三数之和、18.四数之和

四数之和的双指针解法是两层for循环nums[k] + nums[i]为确定值，依然是循环内有left和right下标作为双指针，找出nums[k] + nums[i] + nums[left] + nums[right] == target的情况。首先定义一个unordered_map，key放a和b两数之和，value 放a和b两数之和出现的次数。这组数据，当遍历到第一个-2 的时候，判断下一个也是-2，那这组数据就pass了。遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中。
复制链接

扫一扫