给定未排序数组(可能有重复元素),找出和为给定值的所有二元组元素的下标

题目

给定一个未排序数组(可能有重复元素),找出和为给定值的所有二元组元素的下标
例如:数组{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;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章