Leetcode: Question1--Two Sum

LeetCode上面的难度为easy的一道题,作为进入Leetcode的第一道题,做下来比较顺利,不过跑的结果不好,原因在于算法的复杂度高,下面给出做这道题遇到的问题


问题描述

问题描述


解法

根据题目很容易得出第一种解法,直接暴力求解即可
遍历数组,找出符合要求的两个数,并返回
以下是C++代码

#include <vector>
using namespace std;
class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        for (int i = 0; i < nums.size(); ++i) {
            for (int j = i + 1; j < nums.size(); ++j) {
                if (nums[i] + nums[j] == target) {
                    std::vector<int> v;
                    v.push_back(i);
                    v.push_back(j);
                    return v;
                }
            }
        }

    }
};

这是我做的第一种方法,函数要返回两个数,因此创建了一个 vector 返回
假如在java中,则可以使用

return new int[] { i, j };

但在C++中不能这么使用,原因在于C++中并不存在Garbage Collection,内存的分配和释放必须靠自己
因此返回数组的时候显得十分麻烦

该算法的复杂度是 O(N^2)

另一种解法是使用哈希表,在寻找特定值以及下标的时候可以将复杂度降到O(1),遍历的时候O(N),总复杂度O(N)

#include <vector>
#include <map>
using namespace std;

class Solution {
public:
    static vector<int> twoSum(vector<int>& nums, int target) {
/*      for (int i = 0; i < nums.size(); ++i) {
            for (int j = i + 1; j < nums.size(); ++j) {
                if (nums[i] + nums[j] == target) {
                    std::vector<int> v;
                    v.push_back(i);
                    v.push_back(j);
                    return v;
                }
            }
        }*/
        map<int, int> mymap;

        for (int i = 0; i < nums.size(); ++i) {
            mymap.insert( std::pair<int, int>(nums[i], i) );
        }
        for (int i = 0; i < nums.size(); ++i) {
            int complement = target - nums[i];
            auto it = mymap.find(complement);
            if (it != mymap.end() && it->second != i) {
                   std::vector<int> v;
                    v.push_back(i);
                    v.push_back(it->second);
                    return v;
            }
        }

/*        std::vector<int> v;
        v.push_back(0);
        v.push_back(0);
        return v;
*/

    }
};

这里采用的是C++ STL中的map
结果显示这种做法效率更高
这里写图片描述

猜你喜欢

转载自blog.csdn.net/huangbx_tx/article/details/82528458