给定一个整数数组A和一个整数sum,编写程序找出和为目标值sum的那两个数组元素,并返回它们的数组下标。输入第一行每个分量为数组A的元素,第二行为sum值。 输入样例: 2 7 11 15 9 输出: 0 1
时间: 2023-05-17 20:05:42 浏览: 131
以下是C知道的回答:
可以使用哈希表来解决这个问题,时间复杂度为O(n)。具体实现步骤如下:
1. 创建一个空的哈希表,用于存储数组元素和它们的下标。
2. 遍历数组A,对于每个元素A[i],计算出与目标值sum的差值diff = sum - A[i]。
3. 在哈希表中查找是否存在diff这个键值,如果存在,则说明找到了两个元素的和为sum,返回它们的下标即可。
4. 如果哈希表中不存在diff这个键值,则将当前元素A[i]和它的下标i存入哈希表中。
下面是实现代码:
#include <iostream>
#include <unordered_map>
#include <vector>
using namespace std;
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> hash;
vector<int> res;
for (int i = 0; i < nums.size(); i++) {
int diff = target - nums[i];
if (hash.count(diff)) {
res.push_back(hash[diff]);
res.push_back(i);
return res;
}
hash[nums[i]] = i;
}
return res;
}
int main() {
vector<int> nums = {2, 7, 11, 15};
int target = 9;
vector<int> res = twoSum(nums, target);
for (int i = 0; i < res.size(); i++) {
cout << res[i] << " ";
}
cout << endl;
return 0;
}
输出结果为:0 1
阅读全文