Leetcode Problem 1: Two sum

Description

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.

Example:

Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].

题目的意思是：要在一个数组里面找出两个数，使得他们相加的和等于给定的数，输出两个数的在数组里面对应的索引。

Solution

最基本的思路是穷竭搜索（暴搜），两重循环，时间复杂度O(n^2);这个方法会造成Time Limited Exceed，测试用例是个很大的一维数组。

更好的办法是使用map: 可以达到O(n)的时间复杂度。记忆化搜索，遍历数组时，先在map里面查询target-num（和减去当前的数字），如果没有找到，则将当前数字（做为key）和对应的数组索引（做为value）保存在map中，如果找到了，则直接输出。

这里选择STL里面的unordered_map作为map数据结构的实现。

Code

/** * Leetcode problem list 1: two sum. * Use map algorithm(stl).  * * hellfire(asyncloading#163.com) * Feb 15th, 2016 */#include<iostream>#include<vector>#include<unordered_map>using namespace std;class Solution {  public:    vector<int> twoSum(vector<int>& nums, int target) {      vector<int> output(2);      unordered_map<int, int> umap;            for (int i = 0; i < nums.size(); i ++)      {        if (umap.find(target - nums[i]) != umap.end())        {          output[0] = umap[target - nums[i]];          output[1] = i;        }        else        {          umap.insert(make_pair(nums[i], i));        }        }      return output;     }};int main(int argc, char **argv){  Solution s;  int arr[] = {2, 7, 11, 15};  int target = 13;  vector<int> nums(arr, arr + 3);  vector<int> output = s.twoSum(nums, target);  cout << output[0] << output[1] << endl;}

源代码对应的Github地址: https://github.com/oj-problem/leetcode/blob/master/solution1.cpp

备注

c++语法：不能直接将数组赋值给vector容器，可以采用下面的方式。

  int arr[] = {2, 7, 11, 15};  vector<int> nums(arr, arr + 3);