程式碼隨想錄:兩數之和
題目本身倒是還好,迭代器和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 {};
}
};