LeetCode 1. Two Sum

问题

Given an array of integers, return indices of the two numbers such that they add up to a specific target.

You may assume that each input would have exactly one solution, and you may not use the same element twice.

Example:

Given nums = [2, 7, 11, 15], target = 9,

Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].

分析

如果直接用两个for循环来做的话，非常直接和简单，但是速度很慢。时间复杂度O(n^2),空间复杂度O(1)。
所以使用hash table来做。时间复杂度O(n),空间复杂度O(n)。
C++标准库中没有hash，但是可以使用 unordered_map 来实现，它内部就是一个hash表。
格式：unordered_map

代码

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        int i,j;
        vector<int> result;
        for(i=0; i<nums.size(); i++)
            for(j=i+1; j<nums.size(); j++)
                if(target == nums[i] + nums[j]) 
                {
                    result.push_back(i);
                    result.push_back(j);
                    return result;
                    //上面三行还可以简化为： return {i,j};
                }
         return result;           
    }

    //for循环做实在太low，使用哈希表。C++标准库中没有hash，
    //但是可以使用unordered_map来实现，它内部就是一个hash表。
    vector<int> twoSum(vector<int> &nums, int target)
    {
    //unordered_map<type key, type value>，这里数组值 作key，下标作 value。
      unordered_map<int, int> hash;
      vector<int> result;
      for (int i = 0; i < nums.size(); i++)
      {
          int numberToFind = target - nums[i];
          //if numberToFind is found in map, return them
          if (hash.find(numberToFind) != hash.end()) 
          {/*
            result.push_back(hash[numberToFind] );
            result.push_back(i);    //注意先后顺序，找到说明之前放进去的，所以在前面。      
            return result; */
            //上面三行还可以简化为：
            return {(*(hash.find(numberToFind))).second, i};
          }
          //if number was not found. Put it in the map.
          hash[nums[i]] = i;
       }
       return result;
    }
};

总结

• C++ STL的东西还是要多加熟悉，知道其特性，熟练其操作，并能够灵活使用。
• 参考： 关联容器：unordered_map详细介绍（附可运行代码）