Leetcode- 1.Two sum-对撞指针解法

题目描述:

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

解法:使用先排序+双索引技术(对撞指针),处理细节在于需要返回的是原来的数组索引,方法是在排序前先使用一个额外的vector拷贝一份原来的vector,此过程中针对两个相同元素的索引问题,使用一个bool型变量辅助将两个索引都找到,总的时间复杂度为O(n)+O(nlogn), 具体实现如下:

//by Anyu 20190112
class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> res;
        vector<int>oriNums;
        bool sameFlag=true;
        for(int i=0;i<nums.size();i++)
        {
            oriNums.push_back(nums[i]);
        }
        sort(nums.begin(),nums.end());
        int left = 0;
        int right = nums.size()-1;
        while(left<right)
        {
            if(nums[left]+nums[right]==target)
                break;   
            else if(nums[left]+nums[right]<target)
                left++;
            else
                right--;
        }
        for(int j=0;j<oriNums.size();j++)
        {
            if(oriNums[j]==nums[left]&&sameFlag==true)
            {
                res.push_back(j);
                sameFlag=false;
            }
            else if(oriNums[j]==nums[right])
                res.push_back(j);
        }
        return res;
    }
};

猜你喜欢

转载自blog.csdn.net/tox33/article/details/86349257