Leetcode NO.1 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

本题思路直观,开始做的时候感觉只有easy的难度,但是后面碰上了点问题,花了点时间,不过仍然算是medium中相对简单的。。

下面是代码:

class Solution {
public:
    vector<int> twoSum(vector<int> &numbers, int target) {
        unordered_map<int, int> num_map;
        unordered_map<int, int>::iterator it;
        vector<int> index(2);
        for (int i = 0; i < numbers.size(); ++i) {

        	int tmp = target - numbers[i];
        	it = num_map.find(tmp);
        	if (it != num_map.end()) {
        		index[0] = it->second + 1;
        		index[1] = i + 1;
        	}
        	num_map.insert(pair<int, int>(numbers[i], i));
    	}
    	return index;
    }
};
算法:

基本思路就是在插入到hash_map之前,要先检查是否能找到target-该元素,如果能找到,就直接返回这两个index,如果找不到,则正常插入,这种一步一步的插入方式消除了:

如果target=2,有两个元素的value都是1,如果他们都被插入到multi_map中,如果寻找(2-1)可能返回的index还是第一个1的index..

再上传一个更新后的代码,应该会更快才对,因为找到后及时跳出了:

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> res(2, -1);
        unordered_map<int, int> hashmap;        // key is the number, val is index
        for (int i = 0; i < nums.size(); ++i) {
            auto got = hashmap.find(target - nums[i]);
            if (got != hashmap.end()) {
                res[1] = i + 1;
                res[0] = got->second + 1;
                break;
            }
            hashmap[nums[i]] = i;
        }
        return res;
    }
};



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值