LeetCode 1. 两数之和（哈希）

Michael阿明

发布于 2021-02-20 10:52:54

17500

代码可运行

文章被收录于专栏：Michael阿明学习之路Michael阿明学习之路

运行总次数：0

代码可运行

1. 题目

题目链接：https://leetcode-cn.com/problems/two-sum/

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

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

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

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

2. 解题

1. 暴力解法

每个元素都跟后面的比较，n2 复杂度

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

2. 哈希法

利用unordered_map哈希查找，先建立哈希表O(n)，再查找O(n)

//哈希法
class Solution 
{
public:
    vector<int> twoSum(vector<int>& nums, int target) 
    {
        int i, d;
        vector<int> ans;
        unordered_map<int,int> unomap;//unordered_map是哈希表
        for(i = 0; i < nums.size(); ++i)//建立哈希表
            unomap[nums[i]] = i;//哈希数组下标是值，存的原来的下标
        for(i = 0; i < nums.size(); ++i)
        {
            d = target - nums[i];
            if(unomap.find(d) != unomap.end() && unomap[d]>i)
            {//如果查找到了，且不是同一个元素
                ans.push_back(i);
                ans.push_back(unomap[d]);
                break;
            }
        }
        return ans;
    }
};

优化，只需要一次遍历

class Solution {	// 2020.10.3 
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int,int> m;
        int number;
        for(int i = 0; i < nums.size(); ++i)
        {
            number = target-nums[i];
            if(m.find(number) != m.end())
                return {i, m[number]};
            m[nums[i]] = i;
        }
        return {};
    }
};

3. python3解答

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        dic = {}
        for i in range(len(nums)):
            dic[nums[i]] = i
        for i in range(len(nums)):
            if (target-nums[i]) in dic and dic[target-nums[i]] > i:
                return [i,dic[target-nums[i]]]
        return [-1,-1]