Two Sum[LeetCode]

131 阅读1分钟

题目描述

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。 你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。

示例:

给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

思路1

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> res;
        if(nums.size() == 0){
            return res;
        }
        
        vector<int> temp = nums;
        sort(nums.begin(),nums.end());
        int len = nums.size();
        int start = 0;
        int end = len - 1;
        while(start < end){
            if(nums[start] + nums[end] == target){
                // 这里注意,不要直接返回stat 和 end 作为结果,因为当前nums为排序之后的了,
                // 而我们要返回的为原始的数组的下标
                for(int i = 0; i < temp.size(); i++){
                    if(temp[i] == nums[start] || temp[i] == nums[end]){
                        res.push_back(i);
                        if(res.size() == 2) break;
                    }
                }
                break;
            }else if(nums[start] + nums[end] < target){
                start++;
            }else{
                end--;
            }
        }
        return res;
    }
};

思路2 经典

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