题目:
给一个整数数组,找到两个数使得他们的和等于一个给定的数 target。
你需要实现的函数twoSum
需要返回这两个数的下标, 并且第一个下标小于第二个下标。注意这里下标的范围是 1 到 n,不是以 0
开头。
注意事项
你可以假设只有一组答案。
样例
给出 numbers = [2, 7, 11, 15]
, target = 9
, 返回 [1, 2]
.
解法1:最简单的方法,两重for循环,时间复杂的O(n^2). 提交显示用时10ms.
附代码:
// 解法1 两重for循环 class Solution { public: /* * @param numbers : An array of Integer * @param target : target = numbers[index1] + numbers[index2] * @return : [index1+1, index2+1] (index1 < index2) */ vector<int> twoSum(vector<int> &nums, int target) { // write your code here vector<int> ans; for (int i=0; i<nums.size(); ++i) { for (int j=i+1; j<nums.size(); ++j) { if (nums[i] +nums[j] == target) { return {i+1, j+1}; } } } } };
解法2:用map标记值和下标,一次遍历,如果map里能找到和当前值和为target的值,找到答案。否则插入进map.时间复杂度O(n), 加上map保存的空间复杂度。
// 解法2:O(n) class Solution { public: /* * @param numbers : An array of Integer * @param target : target = numbers[index1] + numbers[index2] * @return : [index1+1, index2+1] (index1 < index2) */ vector<int> twoSum(vector<int> &nums, int target) { // write your code here map<int, int> mp; for (int i=0; i<nums.size(); ++i) { int val = target - nums[i]; int index = i + 1; map<int, int>::iterator iter; iter = mp.find(val); if (iter != mp.end()) { return {iter->second, i+1}; } mp.insert({nums[i], i+1}); } } };
解法3:解法2中的map去掉,直接再vector里找。
// 解法3:O(n) class Solution { public: /* * @param numbers : An array of Integer * @param target : target = numbers[index1] + numbers[index2] * @return : [index1+1, index2+1] (index1 < index2) */ vector<int> twoSum(vector<int> &nums, int target) { // write your code here for (int i=0; i<nums.size(); ++i) { int val = target - nums[i]; int index = i+1; auto iter = find(nums.begin()+i+1, nums.end(), val); if (iter != nums.end()) { return {i+1, iter-nums.begin()+1}; } } } };