leetcode精选题1.两数之和(不是简单的1+2,顺便展示字典和列表的区别)

 

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。 你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。 示例: 给定 nums = [2, 7, 11, 15], target = 9 因为 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]

解答:此题解法挺多的,一开始为了巩固学习enumerate()函数,专门用到enumerate()来解题,代码如下:

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        ls = []
        for i, item in enumerate(nums):
            #print(i, item)
            if (target - item) in nums:
                if (target - item) == item and nums.count(item) == 1:
                    continue
                else:
                    ls.append(i)
        return ls

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

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

这个执行时间吓到我了,所以必须优化,如下

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        d={}
        size=0
        while size < len(nums):
            if target-nums[size] in d:
                if d[target-nums[size]] < size:
                    return [d[target-nums[size]],size]
            else:
                d[nums[size]] = size
            size = size + 1

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

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

从执行时间上,已大大改进,但占用的内存也相比第一个较多。

原因是,第二种方法使用到的是字典,查找速度快,但占用内存大,第一种方法使用列表,查找速度慢,但占用内存小。

这也是很好的阐述python中字典dict和列表list的区别。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值