关关的刷题日记11——Leetcode 1. Two Sum 方法2、3

点击上方“专知”关注获取更多AI知识!

关小刷刷题11 – Leetcode 1. Two Sum 方法2、3

方法2

方法2：先将数组进行排序，设置两个左右指针，若sum>target, 那么右指针左移，sum<target, 左指针右移。sum=target，对应元素的头尾指针就是所求结果。但是有个问题是这时候得到的头尾指针是元素排序后对应的头尾指针，咱们想求的是元素排序前对应的头尾指针，所以一开始要把原来的数组拷贝一份，然后遍历数组，找到对应下标值。时间复杂度为O(nlogn)的方法，还用了额外空间，貌似也不是很好的方法，但是如果数组已经排好序了这种方法就很赞。

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int>output;
        vector<int>temp=nums;
        sort(nums.begin(), nums.end());
        int a=0, b=0;
        for(int i=0, j=nums.size()-1; i<j;)
        {
            if(nums[i]+nums[j]>target)
                j--;
            else if(nums[i]+nums[j]<target)
                i++;
            else
            {
                a=nums[i];
                b=nums[j];
                break;
            }
        }
        for(int i=0; i<temp.size(); i++)
        {
            if(temp[i]==a || temp[i]==b)
                output.push_back(i);
        }
        return output;
    }
};

方法3

方法3：用哈希表，key存数组元素，value存对应的下标值，遍历数组，当查找到target-nums[i]时，此时元素的下标和target-nums[i]的value值就是所求结果。时间复杂度为O (n)的方法，用了额外空间。

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int>output;
        unordered_map<int, int>m;
        for(int i=0;i<nums.size();i++)
        {
            if(m.find(target-nums[i])!=m.end())
            {
                output.push_back(m[target-nums[i]]);
                output.push_back(i);
                break;
            }
            else
                m[nums[i]]=i;
        }
        return output;
    }
};