码迷,mamicode.com
首页 > 其他好文 > 详细

[LeetCode] Two Sum 两数之和

时间:2014-11-29 11:48:14      阅读:184      评论:0      收藏:0      [点我收藏+]

标签:style   blog   io   ar   color   os   sp   for   on   

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 两数之和

标签:style   blog   io   ar   color   os   sp   for   on   

原文地址:http://www.cnblogs.com/grandyang/p/4130379.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!