1. 程式人生 > >Leedcode(1)兩數之和

Leedcode(1)兩數之和

No.1 Two sum
題目
*給定一個整數陣列和一個目標值,找出陣列中和為目標值的兩個數。
你可以假設每個輸入只對應一種答案,且同樣的元素不能被重複利用。
示例:
給定 nums = [2, 7, 11, 15], target = 9
因為 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

解法一 暴力搜尋法

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[j]==target-nums[i])
                    {return {i,j};}
                }
            }
        }
    };

演算法時間複雜度O(n^2),基本思路就是兩層for迴圈巢狀,每個都便利一遍,直到得到合適的為止。

解法二 使用雜湊表
程式碼如下

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