菜鸟扣代码第十七天:leetcode第1题--两数之和

题目描述:

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

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

你可以按任意顺序返回答案。

示例 1:

输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

代码:


class Solution:
    def twoSum(self, nums, target):
        nums_length = len(nums)
        for i in range(0,nums_length):
            num = target - nums[i]
            for j in range(i+1,nums_length):
                if num == nums[j]:
                    return [i,j]


测试:

输入
[2,9,17,15]
19
输出
[0,2]
预期结果
[0,2]

思路:

代码段1:
for i in range(nums_length):
for j in range(nums_length):
if i != j:
if nums[i] + nums[j] == target:
return [i,j]

代码段2:
for i in range(0,nums_length):
num = target - nums[i]
for j in range(i+1,nums_length):
if num == nums[j]:
return [i,j]

代码段1是直接暴力破解的,这样的做法简单粗暴,通俗易懂,但是运行时间较长,进行了重复的计算,例如2+7、7+2.

代码段2在1的基础上进行了改进,两个for循环中,遍历每一个列表元素,查看是否存在与target-num[i]相同的目标元素,如此一来,就省去了重复的计算,提高运行的速度。

猜你喜欢

转载自blog.csdn.net/weixin_51239526/article/details/112850733