LeetCode 1. Two Sum

来源:互联网 发布:win10软件小助手 编辑:程序博客网 时间:2024/06/08 13:50

问题

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详细介绍(附可运行代码)
原创粉丝点击