leetcode 1. 两数之和

问题描述

给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

代码1

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

结果

执行用时 :212 ms, 在所有 C++ 提交中击败了26.53%的用户
内存消耗 :11.5 MB, 在所有 C++ 提交中击败了5.07%的用户

代码2

使用hash表：

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        int i,j,size = nums.size(),tmp;
        vector<int> a(2,0);
        unordered_map<int, int> table;
        for(i = 0; i < size; i++)
        {
            table[nums[i]] = i;
        }
        for(i = 0; i < size; i++)
        {
            tmp = target - nums[i];
            if(table.count(tmp) && table[tmp] != i)
            {
                a[0] = i;
                a[1] = table[tmp];
                break;
            }
        }
        return a;
    }
};

结果：

执行用时 :4 ms, 在所有 C++ 提交中击败了99.76%的用户
内存消耗 :12.6 MB, 在所有 C++ 提交中击败了5.07%的用户

也可以写为一趟循环：

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

结果：

执行用时 :8 ms, 在所有 C++ 提交中击败了96.54%的用户
内存消耗 :12.3 MB, 在所有 C++ 提交中击败了5.07%的用户