Leetcode #1 - Two Sum - Easy

Problem

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.

The return format had been changed to zero-based indices.

Example

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

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

---

Algorithm

整理一下题意：给定一个int数组和一个target，在数组中找到两个元素使得它们的和等于target，找到后返回这两个元素的下标。题目的输入确保有且仅有一组解，元素下标从0开始。

最简单的想法当然是直接两次遍历，时间复杂度为O（n^2）。这其中有个小插曲，如果第四行代码写作vector<int> result;则不能通过，写作vector<int> result（4）;则可以通过。可以猜测后者复杂度比前者略低一些，具体原因尚未在网上找到相关解释，后面查阅了STL的资料再PO上来。

如果想要改进一下代码，可以考虑使用map或者set，时间复杂度为O（nlogn）。利用map建立元素与下标的关系，从而一次遍历完成判断。因为题目规定了有且仅有一组解，所以遍历过程中符合条件即停止，下标组合即为要求输出。

---

Code

//两次遍历版本，O(n^2)，用时799ms
class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> result(2);
        for(int i=0;i<nums.size();i++){
            for(int j=i+1;j<nums.size();j++){
                if(nums[i]+nums[j]==target){
                    result[0]=i;
                    result[1]=j;
                }
            }
        }
    return result;
    }
};

//map版本，O(nlogn)，用时29ms
class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        map<int,int> index;
        vector<int> result(2);
        int i;
        for(i=0;i<nums.size();i++){
            if(index.count(target-nums[i])!=0){
                result[0]=index[target-nums[i]];
                result[1]=i;
            }
            index[nums[i]]=i;
        }
        return result;
    }
};