题:https://leetcode.com/problems/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].
思路
1.扫描一遍nums [i] ,将nusm[i ],i 映射到map,时间复杂度为O(N),空间复杂度为O(N);
2.在map中,扫描一遍target - nums [i],若存在且返回的下标不为i。返回 i 与 上述下标。
##笔记
1.map
声明&定义:
map<int,int> tmap;
查找:
map<int,int>::iterator findres = tmap.find(key);
if(findres != tmap.end()) .....(查找到)
findres->first; //map.key
findres->second; //map.value
Code
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
int ibegin = 0,iend = nums.size()-1;
map<int,int> maprec;
vector<int> res;
for(int i = 0 ;i<nums.size() ; i++) maprec[nums[i]] = i;
for(int i = 0 ;i<nums.size() ; i++){
int complement = target - nums[i];
map<int, int>::iterator findres= maprec.find(complement);
if(findres != maprec.end() && findres->second != i){
res.push_back(i);
res.push_back(findres->second);
return res;
}
}
}
};
第二版
设置Map:key->value。其中 key为 nums[i] (数组中的元素),value为 i(元素对应的下标)。
从头遍历 nums。
若remainder = target - nums[i] 没在 map中,则将该元素及其下标放入map中,map.put(nums[i],i);
否则,那说明 在位置i前的某个元素为nums[i]的 remainder,即 nums[i] + remainder 为target。
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer,Integer> indexForNumMap = new HashMap<>();
for(int i = 0 ;i < nums.length;i++){
int remainder = target - nums[i];
if(indexForNumMap.containsKey(remainder)){
return new int[]{indexForNumMap.get(remainder),i};
}
else
indexForNumMap.put(nums[i],i);
}
return null;
}
}