LeetCode-兩數之和 C++用暴力法與哈希表法分別實現

力扣第一題
話不多說，直接貼代碼

#include <iostream>
#include <vector>
#include <map>
using namespace std;
/**
 * LeetCode
 * 1.兩數之和
 * https://leetcode-cn.com/u/banana798/
 */
class Solution {
public:
    //暴力遍歷法 時間複雜度o(n²)
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> ans;
        for(int i=0; i<nums.size();i++){
            for(int j=i+1;j<nums.size();j++){
                if(nums[i]+nums[j]==target){
                    ans.push_back(i);
                    ans.push_back(j);
                }
            }
        }
        return ans;
    }

    //哈希法 時間複雜度o(n)
    vector<int> twoSum_hash(vector<int>& nums, int target) {
        vector<int> ans;
        map<int,int>hashmap;
        for(int i=0;i<nums.size();i++){
            int tag = target - nums[i];
            if(hashmap[tag]){
                ans.push_back(hashmap[tag]-1);
                ans.push_back(i);
                return ans;
            }
            //防止處理value為0的情況，因為查詢key是否存在時對應value會初始化為0
            hashmap[nums[i]]=i+1;
        }
    }

};


int main() {
    vector<int> nums;
    nums.push_back(2);
    nums.push_back(7);
    nums.push_back(11);
    nums.push_back(15);
    Solution solution;
    vector<int> ans = solution.twoSum_hash(nums, 9);
    cout<<ans[0]<<" "<<ans[1];
}

注意哈希表在探測其key存不存在時，會默認初始化value為0，要避免value為0的情況