LeetCode 1. 两数之和(哈希)

文章目录

    • 1. 题目
    • 2. 解题
      • 1. 暴力解法
      • 2. 哈希法
      • 3. python3解答

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;
    }
};

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]

LeetCode 1. 两数之和(哈希)_第1张图片

你可能感兴趣的:(LeetCode,哈希)