logo头像

往者不可谏,来者犹可追。

LeetCode_problem_1_two_sum

Description

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].


测试用例

输入:3,3 6
输出: 0,1

输入:2,3,3 5
输出: 0,1
下面代码没有通过这个测试,但是A了。题目说结果是确定的,应该就是不考虑这种情况。


思路

  1. 先排序,最快O(nlogn)。再左右夹逼。比较暴力。
  2. 利用hash表,时间复杂度为O(n)。下面代码是基于第二种方法。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
vector<int> twoSum(vector<int>& nums, int target) {
map<int,int> hashMap;
for(int i = 0; i<nums.size(); i ++) hashMap[nums[i]] = i;

//cout<<hashMap.size()<<endl;

vector<int> ans;
for(int i = 0; i<nums.size(); i ++){
const int gap = target - nums[i];
if(hashMap[gap] > 0){
ans.push_back(i);
ans.push_back(hashMap[gap]);
break;
}
}
return ans;
}
官方java答案
1
2
3
4
5
6
7
8
9
10
11
12

public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
int complement = target - nums[i];
if (map.containsKey(complement)) {
return new int[] { map.get(complement), i };
}
map.put(nums[i], i);
}
throw new IllegalArgumentException("No two sum solution");
}
上一篇