LeetCode1.两数之和 双指针法

目录

题目描述

灵机一动 

AC代码

时间以及空间复杂度

测试样例

题目描述

给定一个整数数组 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;
}