题目描述 1
给定一个整数数组 nums 和一个目标值 target,在该数组中找出和为目标值的那两个整数,并返回它们的数组下标。
假设每种输入只会对应一个答案,但是不能重复利用这个数组中同样的元素。
示例
给定 nums = [2, 7, 11, 15],target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9,所以返回 [0, 1]
解答
class Solution(object):
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
for i in range(len(nums)):
for j in range(i+1,len(nums)):
if (nums[i] + nums[j]) == target:
return [i,j]
return None
题目描述 167
给定一个已按照升序排列 的有序数组,找到两个数使得它们相加之和等于目标数。
函数应该返回这两个下标值 index1 和 index2,其中 index1 必须小于 index2。
说明:
返回的下标值(index1 和 index2)不是从零开始的。
假设每个输入只对应唯一的答案,而且不可以重复使用相同的元素。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum-ii-input-array-is-sorted
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
示例
输入: numbers = [2, 7, 11, 15], target = 9
输出: [1,2]
解释: 2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。
解答
解答1
class Solution(object):
def twoSum(self, numbers, target):
"""
:type numbers: List[int]
:type target: int
:rtype: List[int]
"""
for i in range(0,len(numbers)):
rest = target-numbers[i]
numbers_rest = numbers[i+1:]
if rest in numbers_rest:
return[i+1,numbers_rest.index(rest)+i+2]
# numbers_rest.index(rest)+i+1是rest在numbers里的索引
return None
遍历列表,对当前数numbers[i],判断target-numbers[i]是否在剩下的列表元素内。
这一方法时间复杂度太高,for循环遍历列表O(n),判断元素是否在列表内O(n),查找索引O(n),整个算法O(n^3)
解答2 利用双指针
class Solution(object):
def twoSum(self, numbers, target):
"""
:type numbers: List[int]
:type target: int
:rtype: List[int]
"""
# 有序数组
l = 0
r = len(numbers)-1
while l<r:
if numbers[l]+numbers[r]==target:
return [l+1,r+1]
elif numbers[l]+numbers[r]>target:
r -= 1
else:
l += 1
return None