LeetCode15三数之和

• 1.题目链接

https://leetcode-cn.com/problems/3sum/

• 2.题目描述

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

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

示例：

给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

• 3.题目解析

用双指针方法实现，先nums排序，用3个指针操作实现，先用k指向最小数，然后i和j在（k,numsSize）范围内索引，满足nums[k] + nums[i] + nums[j] = 0，条件即可。

当 sum == 0时，记录i，j和k下标，

当sum小于0或者大于0时，分别跳过重复情况。

•    4.代码实现

int comp(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), comp); //数组从小到大排序
    int **res = (int **)malloc(sizeof(int*) * numsSize *numsSize);
    *returnColumnSizes = (int *)malloc(sizeof(int) * numsSize *numsSize);
    int cnt = 0;
    for (int k = 0; k < numsSize - 2; k++) {
        if (nums[k] > 0)                        //nums[k]>0 则三数和一定大于0
            break;
        if (k > 0 && nums[k] == nums[k - 1])    //重复组合，跳过
            continue;
        int i = k + 1;
        int j = numsSize - 1;
        
        while (i < j) {
            int sum = nums[k] + nums[i] + nums[j];
            if (sum == 0) {
                res[cnt] = (int*)malloc(sizeof(int) * 3);
                res[cnt][0] = nums[k];
                res[cnt][1] = nums[i];
                res[cnt][2] = nums[j];
                (*returnColumnSizes)[cnt] = 3;
                while(i < j && nums[i] == nums[++i]);   //nums[i]重复，则跳过
                while(i < j && nums[j] == nums[--j]);   //nums[j]重复，则跳过
                cnt++;
            }
            else if(sum < 0) {
                while(i < j && nums[i] == nums[++i]);
            }
            else {
                while(i < j && nums[j] == nums[--j]);
            }
        }
    }
    *returnSize = cnt; 
    return res;
}

• 5.提交记录