LeetCode攻略之路(一):两数之和

3 篇文章 0 订阅
2 篇文章 0 订阅

题目描述:

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

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

示例:

示例:

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

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

暴力枚举(两重循环,O(n²))

C++写法:

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

执行用时: 204 ms, 在Two Sum的C++提交中击败了6.01% 的用户

内存消耗: 5.6 MB, 在Two Sum的C++提交中击败了0.82% 的用户

Python写法:

class Solution(object):
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        for i,n in enumerate(nums):
            for j in range(i + 1, len(nums)):
                if nums[i] + nums[j] ==target:
                    return i,j

执行用时: 3692 ms, 在Two Sum的Python提交中击败了42.61% 的用户

内存消耗: 7.8 MB, 在Two Sum的Python提交中击败了84.63% 的用户

提交为python3后,结果为:

执行用时: 44 ms, 在Two Sum的Python3提交中击败了99.77% 的用户

内存消耗: 7.9 MB, 在Two Sum的Python3提交中击败了46.97% 的用户

算法二:使用哈希函数:

C++实现:

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

执行用时: 24 ms, 在Two Sum的C++提交中击败了61.48% 的用户

内存消耗: 6.8 MB, 在Two Sum的C++提交中击败了0.82% 的用户

map是基于红黑树实现的,查找复杂度是log(n)

unordered_map内部使用一个大的hash_table,期望查找时间为O(1)。

Python3实现:

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        
        num2idx = dict()
        for i,n in enumerate(nums):
            if target - n in num2idx:
                return num2idx[target - n], i
            else:
                num2idx[n] = i

执行用时: 44 ms, 在Two Sum的Python3提交中击败了99.77% 的用户

内存消耗: 7.9 MB, 在Two Sum的Python3提交中击败了46.97% 的用户

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值