LeetCode18四数之和

代码来源:代码随想录

c 复制代码
/**
 * Return an array of arrays of size *returnSize.
 * The sizes of the arrays are returned as *returnColumnSizes array.
 * Note: Both returned array and *columnSizes array must be malloced, assume caller calls free().
 */
int compare(const void *a,const void *b)
{
    int a1=*(int *)a;
    int b1=*(int *)b;
    return (a1>b1)-(a1<b1);
}
int** fourSum(int* nums, int numsSize, int target, int* returnSize, int** returnColumnSizes)
{
    //1.对数组进行排序
    qsort(nums,numsSize,sizeof(int),compare);
    //2.分配空间,解的数量上限是C(200 4),但经验值为40000,避免过度分配内存
    int **result=(int **)malloc(sizeof(int *)*40000);
    int count=0;//计数器,记录解的数量
    for(int i=0;i<numsSize-3;i++)
    {
        /* 剪枝:如果nums[i]已经大于target且为正数,后面不可能有解 */
        if ((nums[i] > target) && (nums[i] >= 0)) {
            break;
        }
        /* 去重:跳过重复的nums[i] */
        if ((i > 0) && (nums[i] == nums[i - 1])) {
            continue;
        }
        for(int j=i+1;j<numsSize-2;j++)
        {
            /* 剪枝:如果前两数之和已经大于target且nums[j]为正数 */
            if ((nums[i] + nums[j] > target) && (nums[j] >= 0)) {
                break;
            }
            /* 去重:跳过重复的nums[i] */
            if ((j > (i + 1)) && (nums[j] == nums[j - 1])) {
                continue;
            }
            //左指针
            int left=j+1;
            //右指针
            int right=numsSize-1;
            while(left<right)
            {
                long long sum=(long long)nums[i]+nums[j]+nums[left]+nums[right];
                if(sum>target)
                {
                    right--;
                }
                else if(sum<target)
                {
                    left++;
                }
                else
                {
                    //3.找到符合条件的四元组记录到result中
                    int *result_tmp=(int *)malloc(sizeof(int)*4);
                    result_tmp[0]=nums[i];
                    result_tmp[1]=nums[j];
                    result_tmp[2]=nums[left];
                    result_tmp[3]=nums[right];
                    result[count++]=result_tmp;
                    //去重,去掉重复的left和right
                    while ((right > left) && (nums[right] == nums[right - 1])) {
                        right--;
                    }
                    while ((left < right) && (nums[left] == nums[left + 1])) {
                        left++;
                    }
                    right--;
                    left++;
                }
            }
        }
    }
    *returnSize=count;
    int *column=(int *)malloc(sizeof(int)*count);
    for(int i=0;i<count;i++)
    {
        column[i]=4;
    }
    *returnColumnSizes=column;
    return result;
}
相关推荐
万粉变现经纪人1 小时前
如何解决 pip install -r requirements.txt 私有索引未设为 trusted-host 导致拒绝 问题
开发语言·python·scrapy·flask·beautifulsoup·pandas·pip
qq_479875431 小时前
C++ std::Set<std::pair>
开发语言·c++
毕业设计制作和分享1 小时前
springboot150基于springboot的贸易行业crm系统
java·vue.js·spring boot·后端·毕业设计·mybatis
云知谷4 小时前
【C++基本功】C++适合做什么,哪些领域适合哪些领域不适合?
c语言·开发语言·c++·人工智能·团队开发
电子_咸鱼4 小时前
LeetCode——Hot 100【电话号码的字母组合】
数据结构·算法·leetcode·链表·职场和发展·贪心算法·深度优先
仰泳的熊猫4 小时前
LeetCode:785. 判断二分图
数据结构·c++·算法·leetcode
rit84324994 小时前
基于MATLAB实现基于距离的离群点检测算法
人工智能·算法·matlab
l1t5 小时前
DeepSeek辅助利用搬移底层xml实现快速编辑xlsx文件的python程序
xml·开发语言·python·xlsx
C_Liu_6 小时前
C++:list
开发语言·c++
my rainy days6 小时前
C++:友元
开发语言·c++·算法