每日一解 两数之和升级版 输入有序数组

题目 两数之和 II - 输入有序数组

给定一个已按照升序排列 的有序数组,找到两个数使得它们相加之和等于目标数。
函数应该返回这两个下标值 index1 和 index2,其中 index1 必须小于 index2。

说明:
返回的下标值(index1 和 index2)不是从零开始的。
你可以假设每个输入只对应唯一的答案,而且你不可以重复使用相同的元素。

示例:
输入: numbers = [2, 7, 11, 15], target = 9
输出: [1,2]
解释: 2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum-ii-input-array-is-sorted

思路

如果选定一个数,采用二分查找的话,找到符合条件的另一个数,往往需要O(nlogn)的时间复杂度。而之前解决不规则数组的两数之和的时候,采用的是unordered_map的哈希表来实现O(1)的查找,时间复杂度也只有O(n),但空间开销是比较大的。所以理论上最佳的方法就是利用有序数组的特性,实现O(n)时间复杂度和较低的空间开销。
然后我仔细一看才发现双指针本题是可以实现的。刚开始的时候总觉得双指针是会遗漏情况的,仔细一分析发现在有序数组的情况下确实可行。
双指针就是两个指针,分别在数组开始和数组末尾。如果两数之和大于目标了,那么将末尾的指针向前移动,小于目标则将开始的指针向后移动。最开始我想当然地以为可能会出现其中一个指针“走过了”,而导致找不到正确答案。但事实上这种情况不可能发生。例如左指针先走到正确的数字了,右指针还没到,那么和必然大于目标数,导致一直移动右指针直到找到答案,而不会移动左指针导致左侧走过了。
采用双指针的方法时间复杂度为O(n),空间复杂度也只有O(1)。

代码实现

class Solution {
    
    
public:
	vector<int> twoSum(vector<int>& numbers, int target) {
    
    
		vector<int> answer(2);
		int left = 0, right = numbers.size() - 1;
		while (numbers[left] + numbers[right] != target) {
    
    
			numbers[left] + numbers[right] < target ? left++ : right--;
		}
		answer[0] = left + 1;
		answer[1] = right + 1;
		return answer;
	}
};

这里我预设了answer的初始长度,因为虽然是O(1)的空间复杂度,但提交结果太差了,所以略微在这里优化一下。

猜你喜欢

转载自blog.csdn.net/new_DYX/article/details/107456466
今日推荐