时间:2020-05-20
题目地址:https://leetcode-cn.com/problems/two-sum-ii-input-array-is-sorted/description/
题目难度:Easy
题目描述:
给定一个已按照升序排列 的有序数组,找到两个数使得它们相加之和等于目标数。
函数应该返回这两个下标值 index1 和 index2,其中 index1 必须小于 index2。
说明:
返回的下标值(index1 和 index2)不是从零开始的。
你可以假设每个输入只对应唯一的答案,而且你不可以重复使用相同的元素。
思路1:暴力破解
代码段1:超出时间限制
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
result = []
for i in numbers:
temp = numbers.index(i) + 1
for j in numbers[temp:]:
if i + j == target:
r1 = numbers.index(i) +1
r2 = numbers[temp:].index(j) + 1 + r1
result.append(r1)
result.append(r2)
if(len(result) == 2):
return result
总结:
- 直接改的上次的代码,16 /17 个通过测试用例,回头跑了下上次的,也是超时,之前应该是没有这个超长输入的用例
思路2:使用字典
代码段2:通过
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
result = []
temp = {}
j = 1
for i in numbers:
temp[j] = i
j += 1
print(temp)
for o in temp.values():
diff = target - o
if(diff in temp.values()):
# temp = dict(zip(temp.values(), temp.keys()))
index_o = list(temp.keys())[list(temp.values()).index(o)]
print(index_o)
del temp[index_o]
index_diff = list(temp.keys())[list(temp.values()).index(diff)]
print(index_diff)
result.append(index_o)
result.append(index_diff)
return result
总结:
- 字典key是唯一的,输出位置时一开始想用zip()函数交换键值,看到题目中说的不允许重复使用相同的元素,就换成通过value找key的方式。
- 通过value找key : 方式一:使用keys(), values(), index(),见脚本实例;方式二: 自定义函数;方式三:原字典反转 {v:k for k, v in dict.items()}
- 看下优化的
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
visited = {}
for index, number in enumerate(numbers):
if target - number in visited:
return [visited[target-number], index+1]
else:
visited[number] = index + 1
思路3:双指针,思路2完全没有利用升序数组的特性,有点弱
代码段3:通过
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
result = []
o,j = 0, len(numbers)-1
for q in range(0, len(numbers)-1):
if(numbers[o] + numbers[j] < target):
o += 1
if(numbers[o] + numbers[j] > target):
j -= 1
if(numbers[o] + numbers[j] == target):
result.append(o + 1)
result.append(j + 1)
return result
总结:
- 双指针还是不熟啊,很多可以优化的,见下
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
o,j = 0, len(numbers)-1
while(o < j):
if(numbers[o] + numbers[j] < target):
o += 1
if(numbers[o] + numbers[j] > target):
j -= 1
if(numbers[o] + numbers[j] == target):
return [o+1, j+1]