（C语言）教你怎么完成leetcode大门第一关-----《两数之和》

最新推荐文章于 2023-03-29 00:46:46 发布

pursuit14

最新推荐文章于 2023-03-29 00:46:46 发布

阅读量123

点赞数 1

分类专栏： leetcode 文章标签： leetcode c语言算法

本文链接：https://blog.csdn.net/m0_56165538/article/details/120472305

版权

leetcode 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

本文介绍了一种高效算法，通过二分查找结合一次遍历，以时间复杂度低于O(n^2)的方式解决给定整数数组中寻找和为目标值的两个数及其下标的经典问题。作者提供了使用C++实现的示例，包括排序、比较函数和返回结果的处理。

摘要由CSDN通过智能技术生成

问题概叙/*给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出和为目标值 target 的那两个整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。

你可以按任意顺序返回答案。

示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。
示例 2：

输入：nums = [3,2,4], target = 6
输出：[1,2]
示例 3：

输入：nums = [3,3], target = 6
输出：[0,1]

提示：

2 <= nums.length <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
只会存在一个有效答案
进阶：你可以想出一个时间复杂度小于 O(n2) 的算法吗？ */

考虑到直接遍历两次时间复杂度为n2,有点太浪费了。这里是利用二分查找+一次遍历的方法完成题目要求

注意：利用qsort函数排序后，会把原有的nums数组下标弄混，所以这里设置了一个结构体来保存原始数据和下标

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
struct NUMS {
    int num;
    int pos;
};
int cmp(const void* a, const void* b)
{
    return (*(struct NUMS*)a).num - (*(struct NUMS*)b).num;
}
int* twoSum(int* nums, int numsSize, int target, int* returnSize) {
    struct NUMS N[numsSize];
    int i = 0;
    for (i = 0; i < numsSize; i++)//将nums的原始数据存入结构体
    {
        N[i].num = nums[i];
        N[i].pos = i;
    }
    qsort(N, numsSize, sizeof(struct NUMS), cmp);//将结构体N的num元素排序为有序
    for (i = 0; i < numsSize; i++)
    {
        int left = i + 1;
        int right = numsSize - 1;
        while (left <= right)//二分查找
        {
            int mid = (right + left) / 2;
            if (N[i].num + N[mid].num == target)
            {
                int* ret = (int*)malloc(sizeof(int) * 2);
                *returnSize = 2;
                ret[0] = N[i].pos;
                ret[1] = N[mid].pos;
                return ret;//将结果返回给leetcode
                free(ret);//将malloc开辟的动态内存free掉
                ret = NULL;//避免成为野指针
            }
            if (N[i].num + N[mid].num < target)
            {
                left = mid + 1;
            }
            if (N[i].num + N[mid].num > target)
            {
                right = mid - 1;
            }
        }
    }
    *returnSize = 0;
    return NULL;//找元素被成功的情况，返回一个空指针
}