牛客-C语言解法-三数之和

题目链接：

三数之和_牛客题霸_牛客网 (nowcoder.com)

题目简介：

描述

给出一个有n个元素的数组S，S中是否有元素a,b,c满足a+b+c=0？找出数组S中所有满足条件的三元组。

数据范围： 0≤n≤1000，数组中各个元素值满足  ∣val∣≤100

空间复杂度： O(n^2)，时间复杂度  O(n^2)

注意：

1. 三元组（a、b、c）中的元素必须按非降序排列。（即a≤b≤c）
2. 解集中不能包含重复的三元组。

题目解法：

#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <memory.h>
#define NUMLEN(c) (sizeof(c)/sizeof(c[0]))

/*
 *BM54 三数之和
描述
给出一个有n个元素的数组S，S中是否有元素a,b,c满足a+b+c=0？找出数组S中所有满足条件的三元组。

数据范围：0≤n≤1000，数组中各个元素值满足 ∣val∣≤100
空间复杂度：O(n^2)，时间复杂度 O(n^2))

注意：
三元组（a、b、c）中的元素必须按非降序排列。（即a≤b≤c）
解集中不能包含重复的三元组。
*/


/**
 * 
 * @param num int整型一维数组 
 * @param numLen int num数组长度
 * @return int整型二维数组
 * @return int* returnSize 返回数组行数
 * @return int** returnColumnSizes 返回数组列数
 */
//从小排到大
void sort(int *num, int numLen){
    int temp;
    for(int i=0; i<numLen; i++){
        for(int j=i; j<numLen; j++){
            if(num[j] < num[i]){
                temp = num[i];
                num[i] = num[j];
                num[j] = temp;
            }
        }
    }
}

int** threeSum(int* num, int numLen, int* returnSize, int** returnColumnSizes ) {
    // write code here
    int a,bindex,cindex;
    int** retArray = (int**)malloc(sizeof(int*)*numLen*numLen);    
    *returnColumnSizes = (int*)malloc(sizeof(int)*numLen*numLen);   
    // int (*retColSize)[1] =  (int (*)[1])returnColumnSizes;   //该参数一直调不过
    // int retArray[5][3] = {0};       //本地测试
    sort(num, numLen);
    for(int i=0; i<numLen; i++){//从数组头开始遍历a
        a = num[i];             
        bindex = i+1, cindex = numLen-1;    //更新bc索引
        if(a>0)break;               //如果a大于0，此时由于已经升序排列，bc不满足+a等于0，退出循环 
        //双指针
        while(bindex < cindex){     //如果左指针b 大于 右指针c， 则退出
            if(a*(-1) < (num[bindex] + num[cindex]))            //-a < b+c :大边缩小
                cindex--;
            else if(a*(-1) > (num[bindex] + num[cindex]))       //-a > b+c: 小边增大
                bindex++;
            else{                                               //-a = b+c
                retArray[(*returnSize)] = (int*)malloc(sizeof(int)*3);  //动态申请装该结果的空间
                // *(*(returnColumnSizes+(*returnSize))) = 3;
                (*returnColumnSizes)[(*returnSize)] = 3;
                retArray[(*returnSize)][0] = a;                 //将abc装入返回数组，此时已经是升序排列
                retArray[(*returnSize)][1] = num[bindex];
                retArray[(*returnSize)][2] = num[cindex];
                (*returnSize)++;                                //返回数组行数+1
                while(num[cindex] == num[cindex-1])             //如果大边多个元素连续相同，跳到上一个不同元素
                    cindex--;
                cindex--;
            }
        }
        //如果有多个元素连续相同，直接跳到下一个不同元素
        while(a == num[i+1])
            i++;
    }
    // for(int i=0; i<(*returnSize); i++)
    //     printf("\r\n ret[%d]: %d %d %d", i, retArray[i][0], retArray[i][1], retArray[i][2]);
    return retArray;
}

/**************************end******************************************/

int main ()
{
    int retSize = 0;
    int retColSize[10][1] = {0};
    int nums1[] = {-2,0,0,2,2};
    // int nums1[] = {-10,0,10,20,-10,-40};

    int numbersLen = NUMLEN(nums1);
    int** ret = threeSum(nums1, numbersLen, &retSize, (int**)retColSize);
    for(int i=0; i<retSize; i++)
        printf("\r\n retColSize=%d   ret[%d]: %d %d %d", retColSize[i][0], i, ret[i][0], ret[i][1], ret[i][2]);
    return 0;
}