LeetCode 两数之和(hash表)

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:

给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

思路分析: 最简单粗暴的方法就是穷举暴力法。时间复杂度O(n2)

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> result;//结果vector容器
        int length = nums.size();//数组长度
        //穷举两个下标(i < j)
        for (int i = 0; i < length - 1; ++i){
            for (int j = i + 1; j < length; ++j){
                if (nums[i] + nums[j] == target){
                    result.push_back(i);
                    result.push_back(j);
                    return result;
                }
            }
        }
        return result;
    }
};

LeetCode 两数之和(hash表)_第1张图片
当然时间消耗很大。。。
算法优化: 一遍扫描数组,使用hash表标记已经遇到的元素下标,并不断判断是否有元素对符合条件。

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> result;
        int numsSize = nums.size();
        unordered_map<int, int> myMap;//myMap[num]用于记录它在nums数组中的下标
        //一趟扫描数组
        for (int index = 0; index < numsSize; ++index){
            //如果之前已经遇到过target - nums[index]
            if (myMap.count(target - nums[index]) > 0){
                //myMap[target - nums[index]取出target - nums[index]在nums中的下标
                result = {myMap[target - nums[index]], index};
                break;
            }
            else{
                myMap[nums[index]] = index;//否则进行下标标记
            }
        }
        return result;
    }
};

LeetCode 两数之和(hash表)_第2张图片

你可能感兴趣的:(LeetCode,hash表)