题目
给定一个未排序数组(可能有重复元素),找出和为给定值的所有二元组元素的下标
例如:数组{2, 1, 2,3, 4, 0 }, target = 5
答案:{{0, 3},{ 2, 3}, {1,4} }
要求,时间复杂度为O(n),空间复杂度O(n)
思路
网上类似的题目是已排序的数组,通过双指针就可以解决,空间复杂度可以控制在O(1),而未排序的数组需要保存已遍历元素的信息,可以考虑使用map进行存储,这样每次查询的时候时间复杂度可以控制在O(1)。
使用unordered_map<int, vector> m 进行已遍历元素的存储,key 值为已遍历数组元素值,value值为该元素出现的下标次数(因为有重复元素,用vector存储下标)。
遍历到第i个元素**nums[i]的时候,查询target-nums[i]**是否已出现,出现则说明该可以组成题目要求的二元组。
遍历n个元素,时间复杂度为O(n)。
实现
void findpair(vector<int> & nums, int target)
{
unordered_map<int, vector<int>> m;
int n = nums.size();
for(int i = 0; i<n; ++i) {
int sub = target - nums[i];
if(m.count(sub) > 0) {
for(auto iter : m[sub]) {
cout<<iter << " "<< i<< endl;
}
}
if(m.count(nums[i]) == 0) {
m.insert(pair<int, vector<int>> (nums[i], {i}));
}else
m[nums[i]].push_back(i);
}
}
int main()
{
vector<int> nums = {2, 1, 2, 3, 3, 5, 0};
int target = 5;
findpair(nums, target);
return 0;
}