C++刷題-力扣-兩數之和

題目

給定一個整數數組 nums 和一個目標值 target，請你在該數組中找出和爲目標值的那兩個整數，並返回他們的數組下標。

你可以假設每種輸入只會對應一個答案。但是，你不能重複利用這個數組中同樣的元素。

示例:

給定 nums = [3,2,3] target= 6

因爲 nums[0] + nums[2] = 3 + 3 = 6 

所以返回 [0, 2]

解析：

常規解法是通過暴力法，即將數組做循環嵌套進行兩個整數相加即可行到結果，但此種解法得到的時間複雜度爲O(n*n)。想要優化時間複雜度爲O(n)，需要將循環的層數降低爲一層，此時就需要再借助其他的存儲空間，即是用空間換時間。

暴力解法O(n*n)：

vector<int> twoSum(vector<int>& nums, int target) {

for(int i = 0; i < nums.size()  -1; i ++) {

    for(int j = i + 1; j < nums.size(); j ++) {

            if(nums[i] + nums[j] == target) {

                     return vector{i, j};

            }

    }

return vector<int>();

}

優化解法O(n)

vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int> _map;
        for(int i = nums.size() - 1; i >=0 ; i --) {
            int temp = target - nums[i];
            if(_map.find(temp) == _map.end()) {
                _map[nums[i]] = i;
                continue;
            }
            return vector{i, _map[temp]};
        }
        return vector<int>();
    }