首页 > 代码库 > [LeetCode] Two Sum 两数之和

[LeetCode] Two Sum 两数之和

Given an array of integers, find two numbers such that they add up to a specific target number.

The function twoSum should return indices of the two numbers such that they add up to the target, where index1 must be less than index2. Please note that your returned answers (both index1 and index2) are not zero-based.

You may assume that each input would have exactly one solution.

Input: numbers={2, 7, 11, 15}, target=9
Output: index1=1, index2=2

 

这道题一看就知道用暴力搜索肯定没问题,而且猜到OJ肯定不会允许用暴力搜索这么简单的方法,于是去试了一下,果然是Time Limit Exceeded,这个算法的时间复杂度是O(n^2),代码如下:

 

/** * Time Limit Exceeded */class Solution {public:    vector<int> twoSum(vector<int> &numbers, int target) {        vector<int> res;        for (int i = 0; i < numbers.size(); ++i) {            for (int j = i + 1; j < numbers.size(); ++j) {                if (numbers[j] + numbers[i] == target) {                    res.push_back(i + 1);                    res.push_back(j + 1);                }            }        }        return res;    }};

 

那么只能想个O(n)的算法来实现,参考了网友的解法,发现需要用map数据类型,查找复杂度是O(lgn),相当于二分查找法的效率,非常赞。整个实现步骤为:先遍历一遍数组,建立map数据,然后再遍历一遍,开始查找,找到则记录index。代码如下:

 

class Solution {public:    vector<int> twoSum(vector<int> &numbers, int target) {        vector<int> res;        map<int, int> numMap;        for (int i = 0; i < numbers.size(); ++i) {            numMap[numbers[i]] = i;        }        for (int i = 0; i < numbers.size(); ++i) {            int tmp = target - numbers[i];            if (numMap.find(tmp) != numMap.end() && numMap[tmp] != i) {                res.push_back(i + 1);                res.push_back(numMap[tmp] + 1);                break;            }        }        return res;    }};

 

[LeetCode] Two Sum 两数之和