LeetCode-18-4Sum(C语言实现)

/**

 * Return an array of arrays of size *returnSize.

 * Note: The returned array must be malloced, assume caller calls free().

 */

int cmp (const void * a, const void * b){

  return ( *(int*)a - *(int*)b );

}

int** fourSum(int* nums, int numsSize, int target, int* returnSize) {

    qsort(nums, numsSize, sizeof(int), cmp);

    int i, j, l, r, sum;

    int **res = (int**)malloc(sizeof(int*) * (numsSize * (numsSize - 1) * (numsSize - 2) * (numsSize - 3)) / 24);

    //特殊情况处理

    if(numsSize < 4){

        *returnSize = 0;

        return res;

    }

    //一般情况处理

    *returnSize = 0;

    for(i = 0; i < numsSize - 3; ++i){

        //剪枝

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

        if (nums[i]+nums[i+1]+nums[i+2]+nums[i+3] > target)break;

        if (nums[i]+nums[numsSize-3]+nums[numsSize-2]+nums[numsSize-1] < target)continue;

        for(j = i + 1; j < numsSize - 2; ++j){

            //剪枝

            if (j > i+1 && nums[j] == nums[j-1])continue;

            if (nums[i]+nums[j]+nums[j+1]+nums[j+2] > target)break;

            if (nums[i]+nums[j]+nums[numsSize-2]+nums[numsSize-1] < target)continue;

            l = j + 1;

            r = numsSize - 1;

            while(l < r){

                sum = nums[i] + nums[j] + nums[l] + nums[r];

                if(sum == target){

                    res[*returnSize] = (int*)malloc(sizeof(int) * 4);

                    res[*returnSize][0] = nums[i];

                    res[*returnSize][1] = nums[j];

                    res[*returnSize][2] = nums[l];

                    res[*returnSize][3] = nums[r];

                    ++(*returnSize);

                    do{++l;}while(nums[l] == nums[l - 1]);

                    do{--r;}while(nums[r] == nums[r + 1]);

                }

                else if(sum < target){

                    do{++l;}while(nums[l] == nums[l - 1]);

                }

                else{

                    do{--r;}while(nums[r] == nums[r + 1]);

                }

            }

        }

    }

    return res;

}