力扣刷题打卡day1 两数之和1

题目描述

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target 的那 两个 整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。

你可以按任意顺序返回答案。

思路一：暴力解法

• 套两层循环，逐个相加比对，找到恰当结果后返回对应值下标。
• 时间复杂度为$O(n^2)$

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

思路二：使用map

• 使用C++提供的STL模板map，现将nums第一个元素nums[0]存入map，再遍历nums，用target减去nums[i]，在map中寻找对应值，若未找到，将nums[i]存入map，继续遍历，直到找到对应值，返回下标。
• 使用map.find()的时间复杂度为$O(\log n)$，在平均情况下近似为$O(1)$，但是在冲突严重情况下会退化成$O(N)$
• 平均情况下时间复杂度为$O(N)$

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        map<int,int> sol;
        vector<int> res;
        sol[nums[0]]=0;
        for(int i=1;i<nums.size();i++){
            auto it=sol.find(target-nums[i]);
            //判断是否找到结果
            if(it!=sol.end()){
                res.push_back(i);
                res.push_back(it->second);
            }else{
                sol[nums[i]]=i;
            }
        }
        return res;
    }
};

思路三：配合unordered_set

• STL的map使用的是红黑树，查找元素的时间复杂度为$O(\log N)$，将其改为unordered_set后，可以将查找的时间复杂度缩减为$O(1)$

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_set<int> us;
        vector<int> res;
        us.insert(nums[0]);
        for(int i=1;i<nums.size();i++){
            auto it = us.find(target-nums[i]);
            if(it!=us.end()){
                res.push_back(i);
                for(int j=0;j<i;j++){
                    if(nums[j]==*it){
                        res.push_back(j);
                        break;
                    }
                }
            }else{
                us.insert(nums[i]);
            }
        }
        return res;
    }
};