力扣(LeetCode)15. 三数之和(C++/C语言)

一、环境说明

1. 本文是 LeetCode 15题 : 三数之和，使用c语言实现。
2. 快排+对撞指针。
3. 测试环境:Visual Studio 2019。

二、代码展示

C

//排序,双指针向中间对撞
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 * (numsSize-1)/2);//Cn-2种组合
    *returnSize = 0;//返回值大小从0开始
    *returnColumnSizes = (int*)malloc(sizeof(int) * numsSize * (numsSize-1)/2);//三元组最大数量
    for (int i = 0; i < numsSize; ++i) {
   
   
        if (nums[i] > 0) {
   
   //剪枝
            break;
        }
        if (i > 0 && nums[i] == nums[i - 1]) {
   
   //当前数和上一个数重复,结果重复。
            continue;
        }
        int left = i + 1, right = numsSize - 1;//左指针是第二个数，从第一个数的右一位出发向右; 右指针是第三个数，从右边界向左。
        while (left < right) {
   
   //左右不能颠倒
            int sum = nums[left] + nums[right] + nums[i];
            if (sum == 0) {
   
   //保存一个答案
                ans[*returnSize] = (int*)malloc(sizeof(int) * 3);
                (*returnColumnSizes)[*returnSize] = 3;
                ans[*returnSize][0] = nums[i];
                ans[*returnSize][1] = nums[left];
                ans[*returnSize][2] = nums[right];
                (*returnSize)++;
                while ((left < right) && (nums[left] == nums[left+1