Leetcode1 两数之和 C++,Java,Python

Leetcode1 两数之和

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/two-sum/

博主Github：https://github.com/GDUT-Rp/LeetCode

题目：

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

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。

示例 1:

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

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

解题思路：

方法一：哈希（字典 map）

直观想法

在进行迭代并将元素插入到表中的同时，我们还会回过头来检查表中是否已经存在当前元素所对应的目标元素。如果它存在，那我们已经找到了对应解，并立即将其返回。

C++

//
// Created by Lenovo on 2019/3/19.
//

#ifndef LEETCODE_C_PLUSPLUS_LEETCODE1_H
#define LEETCODE_C_PLUSPLUS_LEETCODE1_H

#include <iostream>
#include <vector>
#include <map>
#include <unordered_map>

using namespace std;

class Solution_LeetCode1 {
public:
    vector<int> twoSum(vector<int> &nums, int target) {
        vector<int> targetnum(2);
        map<int, int> temp_map;
        int complement;
        for (int i = 0; i < nums.size(); ++i) {
            complement = target - nums[i];
            if (temp_map.count(complement)) {
                targetnum[0] = i;
                targetnum[1] = temp_map[complement];
                break;
            } else {
                temp_map[nums[i]] = i;
            }
        }
        return targetnum;
    }

    vector<int> faster(vector<int> &nums, int target) {
        vector<int> v;
        std::unordered_map<int, int> hash;
        for (int i = nums.size() - 1; i >= 0; hash[nums[i]] = i, i--) {
            if (hash.find(target - nums[i]) == hash.end())
                continue;
            v.push_back(i);
            v.push_back(hash[target - nums[i]]);
            hash[nums[i]] = i;
        }
        return v;
    }
};

#endif //LEETCODE_C_PLUSPLUS_LEETCODE1_H

Java

class Solution {
    public int[] twoSum(int[] nums, int target) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            int complement = target - nums[i];
            if (map.containsKey(complement)) {
                return new int[] { map.get(complement), i };
            }
            map.put(nums[i], i);
        }
        throw new IllegalArgumentException("No two sum solution");
    }
}

Python

# -*- coding: utf-8 -*-
# @File   : LeetCode1两数之和.py
# @Author : Runpeng Zhang
# @Date   : 2020/5/6
# @Desc   : 给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。


class Solution:
    def twoSum(self, nums, target: int):
        num_dict = {}
        for index, elem in enumerate(nums):
            if target - elem in num_dict:
                return [index, num_dict[target - elem]]
            else:
                num_dict[elem] = index

复杂度分析

时间复杂度：$O(n)$

空间复杂度：$O(n)$