数据结构与算法
双指针、回溯剪枝、图论与搜索
🎨 视觉封面LeetCode Two Sum II - Input array is sorted
Problem
Given an array of integers that is already sorted in ascending order, find two numbers such that they add up to a specific target number.
The function twoSum should return indices of the two numbers such that they add up to the target, where index1 must be less than index2. Please note that your returned answers (both index1 and index2) are not zero-based.
You may assume that each input would have exactly one solution and you may not use the same element twice.
Input: numbers={2, 7, 11, 15}, target=9 Output: index1=1, index2=2
即找到数组中两个元素的和等于给定数字的元素,数组已排序。
Python 实现
# Given an array of integers that is already sorted in ascending order, find two numbers such that they add up to a specific target number.
#
# The function twoSum should return indices of the two numbers such that they add up to the target, where index1 must be less than index2. Please note that your returned answers (both index1 and index2) are not zero-based.
#
# You may assume that each input would have exactly one solution and you may not use the same element twice.
#
# Input: numbers={2, 7, 11, 15}, target=9
# Output: index1=1, index2=2
# author li.hzh
class Solution:
def twoSum(self, numbers, target):
"""
:type numbers: List[int]
:type target: int
:rtype: List[int]
"""
map = {}
index = 1
for val in numbers:
other = target - val
if other in map:
return [map.get(other), index]
map[val] = index
index += 1
def twoSum_without_extra_space(self, numbers, target):
"""
:type numbers: List[int]
:type target: int
:rtype: List[int]
"""
head, tail = 0, -1
cur_sum = numbers[head] + numbers[tail]
while cur_sum != target:
if cur_sum < target:
head += 1
else:
tail -= 1
cur_sum = numbers[head] + numbers[tail]
return [head + 1, len(numbers) + tail + 1]
分析
这里也做了两个解法,解法1,是通用解法。即,不关心数组中元素的排列规律,借助一个map来寻找成对的元素。
但是,题目明确说明数组中的元素已排序,那么显然是有特殊的解法。即解法二,通过首尾元素相加,与给定值比较。如果小于给定元素,显然只能首指针向后移位,再相加比较。如果大于则尾指针前移。每次比较后,都是明确的单向移动,直到最终找到匹配结果。
所有代码开源上传至 GitHub:yummy-code 仓库 · GESP 专题站:GESP WIKI
欢迎加入:C++ GESP/CSP 考级答疑群(688906745) 与 Java/Python交流群(982860385),点击可直接加群。
猜你想读 · 相关文章推荐
LeetCode Minimum Depth of Binary Tree
Problem Given a binary tree, find its minimum depth. The minimum depth is the number of nodes along the shortest path from the root node down to the nearest lea...
LeetCode Path Sum
Problem Given a binary tree and a sum, determine if the tree has a root-to-leaf path such that adding up all the values along the path equals the given sum. For...
LeetCode Path Sum II
Problem Given a binary tree and a sum, find all root-to-leaf paths where each path's sum equals the given sum. For example: Given the below binary tree and sum...
OneCoder (lihongzheshuai)
一个中年人的自留地,记录学习 C++、GESP/NOI、Java、Python 与算法架构的心得体会。本站唯一网址:coderli.com