Leetcode 1. 两数之和

题干

代码

最简单：两次循环 O(n²)

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

进阶：O(n) 使用map来存储nums数组元素以及所对应的下标 注意这个map可以当作哈希表使用的特性

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        map<int,int> m;
        for(int i=0;i<nums.size();i++) {
            if(m.count(target-nums[i]))
                return {m[target-nums[i]],i};
            m[nums[i]]=i;
        }
        return {};
    }
};

从最简单的一步一步来吧~