求无序数组中两元素相加等于特定值的索引

/*******************************************************************************************************
**description:给定一个无序数组和一个数,要求返回数组中两元素相加等于这个数的索引,且第一个索引小于第二个
**            如:Input: a={2, 7, 11, 15},target=9        Output: index1=1, index2=2
********************************************************************************************************/

//如果是直接两层循环进行查找,时间复杂度是O(n^2),太耗时,不采用该方法;
//由于要求返回索引,所以就不能先排序时间复杂度(O(nlogn)),再左右夹逼(时间复杂度O(n)),总的时间复杂度O(nlogn)
//把数组元素和下标存入哈希中,从前遍历元素,然后从哈希中查找data-a[i],哈希查找时间复杂度为O(1),所以总体时间复杂度为O(n)

#include<iostream>
#include<vector>
#include<unordered_map>
using namespace std;
vector<int> twoSum(const vector<int> &arr, int target)
{
    vector<int> result;
    if (arr.size() < 2)
        return result;
    unordered_map<int, int> cache;
    for (int i = 0; i < arr.size(); i++)
    {
        cache[arr[i]] = i;
    }
    for (int i = 0; i < arr.size(); i++)
    {
        int data = target - arr[i];
        if (cache.find(data) != cache.end())
        {
            if (cache[data] != i)
            {
                result.push_back(i);
                result.push_back(cache[data]);
                return result;
            }
        }
    }
    return result;
}

发布了39 篇原创文章 · 获赞 3 · 访问量 3万+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章