给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路一:
一个简单的实现使用了两次迭代。在第一次迭代中,我们将每个元素的值和它的索引添加到表中。然后,在第二次迭代中,我们将检查每个元素所对应的目标元素(target - nums[i])
是否存在于表中。注意,该目标元素不能是 nums[i] 本身!
优化:
事实证明,我们可以一次完成。在进行迭代并将元素插入到表中的同时,我们还会回过头来检查表中是否已经存在当前元素所对应的目标元素。如果它存在,那我们已经找到了对应解,并立即将其返回。
哈希表,它支持以近似恒定的时间进行快速查找。我用“近似”来描述,是因为一旦出现冲突,查找用时可能会退化到 O(n)。但只要你仔细地挑选哈希函数,在哈希表中进行查找的用时应当被摊销为 O(1)。
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> res;
int len = nums.size();
if (len < 2)
return res;
unordered_map<int, int> hashTable;
for (int i = 0; i < len; i++) {
int complement = target - nums[i];
if (hashTable.find(complement) != hashTable.end()) {
res.push_back(hashTable[complement]);
res.push_back(i);
break;
} else
hashTable[nums[i]] = i;
}
return res;
}
};
思路二:
先排序然后查找。(设置一个pair<int, int> pii;
first保存第i个元素的值,second保存最初的下标i。之后对first的值进行排序。)
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> res;
int len = nums.size();
if (len < 2)
return res;
vector<pair<int, int>> assist;
for(int i = 0; i < len; i++) {
pair<int, int> pii;
pii.first = nums[i];
pii.second = i;
assist.push_back(pii);
}
sort(assist.begin(), assist.end());
int left = 0;
int right = len - 1;
while(left < right && right < len) {
int tmp = assist[left].first + assist[right].first;
if (tmp == target) {
res.push_back(assist[left].second);
res.push_back(assist[right].second);
break;
} else if (tmp > target) {
right--;
} else {
left++;
}
}
return res;
}
};