1. 程式人生 > 其它 >Leetcode 兩數之和

Leetcode 兩數之和

技術標籤:Leetcodeleetcode雜湊表

兩數之和題目連結
題目大意
給定一個整數陣列 n u m s nums nums 和一個目標值 t a r g e t target target,請你在該陣列中找出和為目標值的那 兩個 整數,並返回他們的陣列下標。
你可以假設每種輸入只會對應一個答案。但是,陣列中同一個元素不能使用兩遍。
樣例解釋
給定 n u m s = [ 2 , 7 , 11 , 15 ] , t a r g e t = 9 nums = [2, 7, 11, 15], target = 9 nums=[2,7,11,15],target=9

因為 n u m s [ 0 ] + n u m s [ 1 ] = 2 + 7 = 9 nums[0] + nums[1] = 2 + 7 = 9

nums[0]+nums[1]=2+7=9
所以返回 [ 0 , 1 ] [0, 1] [0,1]

cpp
在這裡插入圖片描述

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) return {i,j};
        return
{}; } };

O ( n ) O(n) O(n)做法
開個雜湊表存數,每次查詢往前面找,如果前面存在一個數等於 t a r g e t target target - n u m s [ i ] nums[i] nums[i],則返回它們的下標,雜湊表的插入和查詢的時間複雜度是 O ( 1 ) O(1) O(1) 這樣去查詢時間複雜度是 O ( n ) O(n) O(n)
C p p Cpp Cpp 中的 m a p map map 底層實現是平衡樹,時間複雜度是 O ( l o g n ) O(logn) O(logn), C p p 11 Cpp11

Cpp11 版本中有個 u n o r d e r e d _ m a p unordered\_map unordered_map 底層實現是雜湊表,時間複雜度是 O ( 1 ) O(1) O(1)

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