程式碼隨想錄:兩數之和

huigugu發表於2024-11-25

程式碼隨想錄:兩數之和

題目本身倒是還好,迭代器和map的使用方法倒是值得記錄,迭代器寫在另一篇了。

map的每一個元素都是一個pair<key,value>,在插入的時候這麼些

visited.insert(std::pair<int, int>(nums[i], i));

或者用make_pair自動識別鍵值對的變數型別

visited.insert(std::make_pair(nums[i], i));

在用迭代器遍歷時,迭代器的it類似於指標,所以用→來讀取指標對應的內容即可,完整程式碼如下

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int> visited;
        for (int i = 0; i < nums.size(); i++) {
            int complement = target - nums[i];
            auto it = visited.find(complement);
            if (it != visited.end()) {
                return {it->second, i};
            }
            visited[nums[i]] = i;
        }
        return {};
    }
};

相關文章