Leetcode15三数之和

15三数之和

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

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

示例：

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

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

来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/3sum 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

Solution 1

快排+双指针
时间复杂度：O(n2)
执行用时：188 ms
内存消耗：31.6 MB

int comp(const void *a, const void *b)
{
    return *(int *)a - *(int *)b;
}

int **threeSum(int *nums, int numsSize, int *returnSize, int **returnColumnSizes)
{
    int MAX_RETURN_SIZE = numsSize * numsSize;
    *returnSize = 0;
    *returnColumnSizes = malloc(sizeof(int) * MAX_RETURN_SIZE);
    int **result = malloc(sizeof(int *) * MAX_RETURN_SIZE);
    int *arr;
    int i = 0, j = 0, k = 0;
    qsort(nums, numsSize, sizeof(int), comp);
    for (i = 0; i < numsSize; i++)
    {
        if (i > 0 && nums[i] == nums[i - 1])
            continue;
        j = i + 1;
        k = numsSize - 1;
        while (j < k)
        {
            if (nums[i] + nums[j] + nums[k] < 0)
                j++;
            else if (nums[i] + nums[j] + nums[k] > 0)
                k--;
            else //(nums[i]+nums[j]+nums[k] == 0)
            {
                arr = malloc(3 * sizeof(int));
                arr[0] = nums[i];
                arr[1] = nums[j];
                arr[2] = nums[k];
                (*returnColumnSizes)[*returnSize] = 3;
                result[(*returnSize)++] = arr;
                while (j < k && nums[j + 1] == nums[j++])
                    ;
                while (j < k && nums[k - 1] == nums[k--])
                    ;
            }
        }
    }
    return result;
}