LeetCode 1. Two Sum  

135 阅读1分钟

LeetCode 1. Two Sum  

给定一个整数数组 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) 的算法吗?

算法1

(暴力枚举) O(n2)

暴力枚举方法很简单:两重循环枚举下标 i,j,然后判断 nums[i]+nums[j] 是否等于 target。

时间复杂度

由于有两重循环,所以复杂度是 O(n2).

C++ 代码

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target)
    {
        vector<int> res;
        for (int i = 0; i < nums.size(); i++)
        {
            for (int j = 0; j < i; j++)
            {
                if (nums[i] + nums[j] == target)
                {
                    res = vector<int>({j, i});
                    break;
                }
            }
            if (res.size() > 0) break;
        }
        return res;
    }
};

算法2

(哈希表) O(n)
使用C++中的哈希表——unordered_map<int, int> hash.
循环一遍 nums 数组,在每步循环中我们做两件事:
判断 target−nums[i] 是否在哈希表中;
将 nums[i] 插入哈希表中;
解释:由于数据保证有且仅有一组解,假设是 i,j ,则我们循环到 j 时,nums[i]一定在哈希表中,且有 nums[i]+nums[j]==target, 所以我们一定可以找到解。

时间复杂度:

由于只扫描一遍,且哈希表的插入和查询操作的复杂度是 O(1) ,所以总时间复杂度是 O(n).

C++ 代码:

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target)
    {
        vector<int> res;
        unordered_map<int,int> hash;
        for (int i = 0; i < nums.size(); i ++ )
        {
            int another = target - nums[i];
            if (hash.count(another))
            {
                res = vector<int>({hash[another], i});
                break;
            }
            hash[nums[i]] = i;
        }
        return res;
    }
};