Algorithms(一)Two Sum

题目:

Two Sum

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, and you may not use the same element twice.

Example:

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

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

代码:

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
	    unordered_map<int, int> hash;
	    vector<int> result;
	    for (int i = 0; i < nums.size(); i++) {
		    int num_find = target - nums[i];

		    if (hash.count(num_find) != 0) {
			    result.push_back(hash[num_find]);
			    result.push_back(i);
			    return result;
		}

		hash[nums[i]] = i;
	}
	return result;
    }
};

思路:
一开始我设计的算法是两个for循环,对nums中的每一个元素与其他元素相加,看结果是否等于target。这当然是最容易想到的方法,但是这个方法的复杂度为O(n^2),开销太大,是个很差劲的算法。
当然我之后又做了一些改进,但基本还是绕着n^2打转,没太大的变化。上面代码中的算法是我目前找到最好的。它用了unordered_map容器,将nums中元素的值和元素下标存入hash中,对nums中每个元素与target做减法,查找目标的number是否在hash中,若在,则输出成result;若不在,则继续循环。
这个算法极大地降低了的复杂度,直接从O(n^2)降到O(n),确实值得学习。

补充知识:

1.使用unordered_map查找元素是否存在的方法:

若有unordered_map<int, int> mp;查找x是否在map中

方法1:若存在 mp.find(x) != mp.end()

方法2:若存在 mp.count(x) = 0

2.vetor.push_back 在数组的最后添加一个数据

参考文章:

c++中vector的用法详解
unordered_map的用法


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值