先看问题描述
题目给定一个整数型数组以及一个目标和,要求找出数组中和为目标和的整数对(两个整数相互独立)。特别的,题目假定输入限制了数组中有切仅有一对整数答案。
首先想到的办法肯定是对于数组进行遍历,在这一层遍历的基础上再建立一个向后的遍历以查找答案,它的基本思想是这样的:
in array a1, a2, ..., an
for i from 1 to n-1
for j from i+1 to n
if (ai + aj == target)
return <ai, aj>
这样的做法很明显效率较差,两层循环造成了O(n²)的复杂度。
因此,我利用了STL的map,它有find函数可以较快地找到我需要的数据:
1. 将数组遍历,将具体值与其下边组成数据对放入map中
2. 再次遍历数组,对于每一个整数,在map中查找target与该整数的差。如果存在这个数且相互独立(两个数下标不同),则找到了答案。
具体代码如下
vector<int> twoSum(vector<int>& nums, int target) {
map<int, int> findAnswer;
map<int, int>::iterator findB;
for (int i = 0; i < nums.size(); i++)
findAnswer[nums[i]] = i;
for (int i = 0; i < nums.size(); i++) {
int temp = target - nums[i];
findB = findAnswer.find(temp);
if (findB != findAnswer.end() && findB->second != i) {
vector<int> answer(2);
answer[1] = findB->second;
answer[0] = i;
return answer;
}
}
}