Leetcode两数之和 C++实现

发布于:2024-08-14 ⋅ 阅读:(116) ⋅ 点赞:(0)

167. 两数之和 II - 输入有序数组

题目:给你一个下标从 1 开始的整数数组 numbers ,该数组已按非递减顺序排列,请你从数组中找出满足相加之和等于目标数 target 的两个数。如果设这两个数分别是 numbers[index1] numbers[index2] ,则 1 <= index1 < index2 <= numbers.length 。

以长度为 2 的整数数组 [index1, index2] 的形式返回这两个整数的下标 index1 和 index2 

你可以假设每个输入只对应唯一的答案,而且你不可以重复使用相同的元素。

你所设计的解决方案必须只使用常量级的额外空间。

算法:由于提供的数组为递增数组,使用双指针策略,左指针 left 置于最左侧,右指针 right 置于最右侧,如果两数之和 sum 大于 target ,则右指针左移,反之左指针右移。

class Solution {
public:
    vector<int> twoSum(vector<int>& numbers, int target) {
        int left = 0;
        int right = numbers.size()-1;
        while (left < right) {
            int sum = numbers[left] + numbers[right];
            if (sum < target)   left++;//左指针右移
            else if (sum > target)    right--;//右指针左移
            else    return vector<int>{left+1, right+1};//满足条件,返回
        }
        return vector<int>{-1, -1};//没有可行解,返回
    }
};

15. 三数之和

题目:给你一个整数数组 nums ,判断是否存在三元组 [ nums[ i ] , nums[ j ] , nums[ k ] ] 满足 i  != ji  != k j  != k ,同时还满足 nums[ i ] + nums[ j ] + nums[ k ] == 0 。请你返回所有和 为 0 且不重复的三元组。

注意:答案中不可以包含重复的三元组。

算法:先对输入数组 nums 进行排序,使用双指针策略,一共规定三个指针 : i ,j ,k ,定住最左边的 i 指针,使得 - nums[ i ] = nums[ j ] + nums[ k ]。 i 的范围是从 0 到 n - 2 ,是因为要留出右面两个指针的位置。因为规定答案中不可以包含重复的三元组,所以要跳过 i 的重复数字。在这里可以对算法进行优化,如果 i 指针所指的这个数加上它右面相邻的两个数仍 > 0 ,就没有继续的必要了。如果 i 指针所指的这个数加上 nums 中最大的两个数的两个数仍 < 0 ,证明此时 i 所指的数字不可能是正确结果,continue

在输出结果时,要注意 jk 也要跳过重复数字。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        ranges::sort(nums);
        vector<vector<int>> ans;
        int n = nums.size();
        for(int i = 0;i < n - 2;i++){
            int x = nums[i];
            if(i && x == nums[i - 1])   continue;//跳过重复数字
            if(x + nums[i + 1] + nums[i + 2] > 0)   break;//优化
            if(x + nums[n - 2] + nums[n - 1] < 0)   continue;//优化
            int j = i + 1,k = n - 1;
            while(j < k){
                int sum = x + nums[j] + nums[k];
                if(sum > 0)   k--;
                else if(sum < 0)  j++;
                else{
                    ans.push_back({x,nums[j],nums[k]});
                    for(j++; j < k && nums[j] == nums[j - 1]; j++);//跳过重复数字
                    for(k--; j < k && nums[k] == nums[k + 1]; k--);//跳过重复数字
                }
            }
        }
        return ans;
    }
};