C++ 笔试每日一题 (two sum)

 

为了对运行时间复杂度进行优化,我们需要一种更有效的方法来检查数组中是否存在目标元素。如果存在,我们需要找出它的索引。保持数组中的每个元素与其索引相互对应的最好方法是什么?哈希表

通过以空间换取速度的方式,我们可以将查找时间从 O(n) 降低到 O(1)。哈希表正是为此目的而构建的,它支持以 近似 恒定的时间进行快速查找。我用“近似”来描述,是因为一旦出现冲突,查找用时可能会退化到 O(n)。但只要你仔细地挑选哈希函数,在哈希表中进行查找的用时应当被摊销为 O(1)

一个简单的实现使用了两次迭代。在第一次迭代中,我们将每个元素的值和它的索引添加到表中。然后,在第二次迭代中,我们将检查每个元素所对应的目标元素(target - nums[i]target−nums[i])是否存在于表中。注意,该目标元素不能是 nums[i]nums[i] 本身!

#include <iostream>
#include <string>
#include <unordered_map>
#include "math.h"
#include <vector>
#include  <unordered_set>
using namespace std;
class Solution{
public:
    vector<int> twoSum(const vector<int> &nums,int target){
      unordered_map<int,int> mapping;

      vector<int> result;
    /*第一次迭代吧,把每个严肃的值和他的索引添加到表中*/
      for(int i=0;i<nums.size();i++){
        mapping[nums[i]]=i;  //用哈希表存储每个数对应的下标
      }

    /*第二次迭代,检查每个元素所对应的目标元素(target-num[i])是否在表中,注意该元素不能为其本身*/
      for(int i=0;i<nums.size();i++){
          const int gap=target-nums[i];
          //遍历哈希表并且不在哈希表中,jiu
          if(mapping.find(gap)!=mapping.end()&&mapping[gap]!=i){
              result.push_back(i);
              result.push_back(mapping[gap]);
              break;
          }
      }
      return result;
    }
};

int main(int argc, char *argv[])
{
    Solution Solu;
    int test[] ={2,7,2,1};
    vector<int>  vec(test,test+sizeof(test)/sizeof(int));
    int target=4;
    for(int i=0;i<Solu.twoSum(vec,target).size();i++){
        cout<<Solu.twoSum(vec,target)[i]<<endl;
    }
    return 0;
}
发布了102 篇原创文章 · 获赞 26 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/weixin_42145502/article/details/103927488
今日推荐