欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

LeetCode1:两数之和 (C、C++、python实现)

程序员文章站 2022-03-08 10:25:03
...

leetcode1

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]

算法

采用hash_map
时间复杂度 O(n)、空间复杂度O(n)

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

C++

采用STL库中map实现
map创建: map<int,int> numsMap; (第二个为坐标)
map插入: numsMap.insert(make_pair(nums[i], i));
map查找: map<int,int>::iterator it = numsMap.find(target_2);
     //取第二个元素为坐标it->second
     //查找不到时: it == numsMap.end()
vector创建: ector vecnum = {3,3};

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

using namespace std;
vector<int> twoSum(vector<int>& nums, int target);

int main(int argc, const char * argv[]) {
    // insert code here...
    vector<int> vecnum = {3,3};
    int target = 6;
    twoSum(vecnum, target);
    return 0;
}
vector<int> twoSum(std::vector<int>& nums, int target) {
    map<int,int> numsMap;
    map<int,int>::iterator it;
    vector<int> result;
    for(int i=0;i<nums.size();i++){
        int target_2 = target - nums[i];
        it = numsMap.find(target_2);
        if(it != numsMap.end()){
            result = {it->second,i};
            cout << it->second<<i;
        }
        numsMap.insert(make_pair(nums[i], i));
    }
    return result;
}

C

python

def twoSum( nums, target):
    map = {}
    for index,num in enumerate(nums):
        target2 = target - nums[index];
        if target2 in map:
            return [map[target2],index];
        map[num] = index
    return   None

if __name__ == "__main__":
    res = twoSum(nums=[3,3],target=6)
    print(res)
相关标签: LeetCode 题解