LeetCode刷题——剑指 Offer 57. 和为s的两个数字

题目

输入一个递增排序的数组和一个数字s，在数组中查找两个数，使得它们的和正好是s。如果有多对数字的和等于s，则输出任意一对即可。

示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[2,7] 或者 [7,2]

示例 2：

输入：nums = [10,26,30,31,47,60], target = 40
输出：[10,30] 或者 [30,10]

限制：

1 <= nums.length <= 10^5
1 <= nums[i] <= 10^6

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/he-wei-sde-liang-ge-shu-zi-lcof

分析

将所有的数存入字典当中，如果容器中含有和为target的两个数a和b那么，当遍历到a时检查target-a即获得b的值，如果target-a在字典中则说明一定存在b使得a和b之和为target

代码

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        set<int> s;
        for(int i=0; i < nums.size(); i++){
            s.insert(nums[i]);
        }
        vector<int> ans;
        for(int i=0; i < nums.size(); i++){
            if(s.count(target - nums[i])){
                ans.push_back(nums[i]);
                ans.push_back(target - nums[i]);
                break;
            }
        }
        return ans;
    }
};