[leetcode]1.Two Sum #array

题目描述

https://leetcode.com/problems/two-sum/
给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的两个整数的index。你可以假设每个输入都只有一个解决方案，并且不能两次使用相同的元素

样例

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

题目解答

法1

最先想到的暴力，两个for循环，时间复杂度为O（n²），c++。

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        int i,j,len=nums.size();
        int flag=0;
        vector<int> res;
        for(i=0;i<len;i++){
            for(j=i+1;j<len;){
                if((nums[i]+nums[j])==target){
                    flag=1;
                    break;
                }else{
                    j++;
                }       
            }
            if(flag==1){
                break;
            }
        }
        res.push_back(i);
        res.push_back(j);
        return res;     
    }
};

法2

思路

一般来说，要减少时间复杂度，就要空间来换。这里使用hashmap，O（n）。第一遍遍历时，一次只遍历一个数字，那么另一个数字就先存起来，使用hashmap建立值和其index间映射。第二次遍历数组nums时，用target减去当前遍历到的值，就是另一个需要找的值，在hashmap中查是否存在，且要判断找到的值不是本身。如target是4，遍历到2，那么在hashmap中找到的2的index不能跟现在index相同。

相关

m.count()，返回被查找元素的个数。如果有，返回1；否则，返回0。注意，map中不存在相同元素，所以返回值只能是1或0。

m.find()，返回被查找元素的位置，没有则返回map.end()。

代码

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

简化版：只一次遍历nums，遍历时，查找target-nums[i]是否在m里，若在，则return，否则将该数存在map里继续查找。

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