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

来源:互联网 发布:mac pro 支持pcie ssd 编辑:程序博客网 时间:2024/05/16 17:11
/*********************************************************************************************************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;}

阅读全文
0 0