力のボタン----3は0を数えます

2078 ワード

リンク:https://leetcode-cn.com/problems/3sum
 
n個の整数を含む配列numsを与えて、numsの中に3つの要素a,b,cが存在するかどうかを判断して、a+b+c=0にしますか?条件を満たし、繰り返さないすべての三元グループを見つけてください.
注意:答えに重複する三元グループは含まれてはいけません.
 
例:
与えられた配列nums=[−1,0,1,2,−1,−4],
要求を満たす三元群の集合は,[[−1,0,1],[−1,−1,2]]である.
 
int CompareByIncrease(const void* a, const void* b)
{
    return *(int*)a - *(int*)b; //         
}

int** threeSum(int* nums, int numsSize, int* returnSize, int** returnColumnSizes){

    *returnSize = 0;

    if( numsSize < 3 )
    {
        return NULL;
    }
    
    int cur = 0;
    int low = cur + 1;
    int high = numsSize - 1;
    

    int** returnArray = (int**)malloc(sizeof(int*) * (numsSize)*(numsSize));
    *returnColumnSizes = (int*)malloc(sizeof(int) * (numsSize)*(numsSize)); 

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

    while((nums[cur] <= 0) && (cur + 1 < numsSize - 1)) 
    {
 
        low = cur + 1;
        high = numsSize - 1;
        
        while(low < high)
        {
            if(0 == (nums[low] + nums[high] + nums[cur]))
            {
                returnArray[*returnSize] = (int*)malloc(sizeof(int) * 3); //       ,           
                (*returnColumnSizes)[*returnSize] = 3; //     ,         
                returnArray[*returnSize][0] = nums[cur];
                returnArray[*returnSize][1] = nums[low];
                returnArray[*returnSize][2] = nums[high];
                (*returnSize)++;
                
                //  low high  ,        ,       
                while( (nums[low] == nums[++low]) && (low < high) ); //     
                while( (nums[high] == nums[--high]) && (low < high) ); //     
            }
            else if(0 < (nums[low] + nums[high] + nums[cur]))
            {
                high--;
            }
            else
            {
                low++;
            }
        }
    
        while( (nums[cur] == nums[++cur]) && (cur + 1 < numsSize - 1) );  
    }
    return returnArray;
}