LeetCode 第一题 两数之和

题目描述

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

方法一:暴力法

暴力法很简单,遍历每个元素 xx,并查找是否存在一个值与 target - xtarget−x 相等的目标元素。

int * twoSum(int * nums, int numSize, int target)
{
    if ((nums == NULL) || (numSize <= 1))
    {
        return NULL;
    }
    int * result = NULL;
    for (int i = 0; i < numSize; i++)
    {
        for (int j = i + 1; j < numSize; j++)
        {
            if (nums[i] + nums[j] == target)
            {
                result = (int *)malloc(sizeof(int) * 2);
                if (result == NULL)
                {
                    printf("malloc error\n");
                    return;
                }
                *result = i;
                *(result + 1) = j;
                return result;
            }
        }
    }
    return result;
}

方法二:使用哈希表代替数组元素遍历

以空间换取速度的方式,可以将查找时间从 O(n)O(n) 降低到 O(1)O(1)。哈希表正是为此目的而构建的,它支持以 近似 恒定的时间进行快速查找。

注意,该目标元素不能是 nums[i]nums[i] 本身!
vector<int> twoSum(vector<int>& nums, int target)
{
    int length = nums.size();
    if (length <= 1)
    {
        return{ -1, -1 };
    }
    map<int, int> cunt;
    vector<int> result(2, -1);
    for (int i = 0; i < length; i++)
    {
        cunt.insert(map<int, int>::value_type(nums[i], i));
    }
    for (int i = 0; i < length; i++)
    {
        if ((cunt.count(target - nums[i]) > 0) && (cunt[target - nums[i]] != i))
        {
            result[0] = i;
            result[1] = cunt[target - nums[i]];
            break;
        }
    }
    return result;
}

方法三:哈希表法,仅遍历一次

一边遍历数组,一边检测哈希表中是否有满足 (target - nums[i])的元素

vector<int> twoSum(vector<int>& nums, int target) {
        int length = nums.size();
        vector<int> result(2, -1);
        if(length <= 1)
        {
            return result;
        }
        map<int, int> cunt;
        for(int i = 0; i < length; i++)
        {
            if(cunt.count(target - nums[i]) > 0)
            {
                result[0] = cunt[target - nums[i]];
                result[1] = i;
                break;
            }
            cunt[nums[i]] = i;
        }
        return result;
    }

猜你喜欢

转载自www.cnblogs.com/Hijack-you/p/11973664.html
今日推荐