求無序數組中兩元素相加等於特定值的索引

/*******************************************************************************************************
**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萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章