目录
题目描述
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
示例 2:输入:nums = [3,2,4], target = 6
输出:[1,2]
示例 3:输入:nums = [3,3], target = 6
输出:[0,1]
提示:
2 <= nums.length <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
只会存在一个有效答案
灵机一动
其实如果不是题目最下面有个欠打(误)的:
你可以想出一个时间复杂度小于
O(n²)
的算法吗?
我就会跟大多数一样两个for循环走天下(逃
看到这句话之后想到用哈希图(O(n))和双指针(O(nlogn))两个方法,都小于O(n²)
哈希图的方法很多大佬都解析过了,我今天分享一下双指针的方法。
AC代码
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> copy = nums; // 对原始数组做好备份,以便最后找到原来的下标
sort(nums.begin(), nums.end()); // 对数组进行排序
// 双指针,一个从前往后,一个从后往前
// 因为排序过后必然是一个下标小(代表数值小)
// 和一个下标大(代表数值大)的两个数加起来等于target,两个数相等也能考虑到
int i = 0;
int j = nums.size() - 1;
while (nums[i] + nums[j] != target) // 如果和等于 target,说明找到了答案
if (nums[i] + nums[j] > target) j--; // 如果和大于target,说明右指针指向的元素太大,需要左移右指针
else i++; // 如果和小于target,说明左指针指向的元素太小,需要右移左指针
// 以下是寻找对应的元素的原来的下标
int p = 0;
int q = nums.size() - 1;
for (; p < nums.size(); p++)
if (copy[p] == nums[i])
break;
for (; q >= 0; q--)
if (copy[q] == nums[j] && q != p) // q!=p防止一个元素用两次
break;
return {p,q};
}
};
时间以及空间复杂度
时间复杂度:O(nlongn),执行时间8ms(超过91.6%的人)。
空间复杂度:O(n),其中n为数组元素的数量,内存消耗10.1M。
测试样例
int main() {
Solution s;
vector<int> nums1 = {2, 7, 11, 15};
int target1 = 9;
vector<int> result1 = s.twoSum(nums1, target1);
cout << "Example 1: ";
for (int i = 0; i < result1.size(); i++)
cout << result1[i] << " "; // expected output : 0 1
cout << endl;
vector<int> nums2 = {3, 2, 4};
int target2 = 6;
vector<int> result2 = s.twoSum(nums2, target2);
cout << "Example 2: ";
for (int i = 0; i < result2.size(); i++)
cout << result2[i] << " "; // expected output : 1 2
cout << endl;
vector<int> nums3 = {3, 3};
int target3 = 6;
vector<int> result3 = s.twoSum(nums3, target3);
cout << "Example 3: ";
for (int i = 0; i < result3.size(); i++)
cout << result3[i] << " "; // expected output : 0 1
cout << endl;
return 0;
}