leetcode/1.cpp at master · Yashbhatt7/leetcode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
#include<iostream>
#include<vector>
#include<unordered_map>

std::vector<int> twoSum(std::vector<int>& nums, int target) {
    std::unordered_map<int, int> freq;
    std::vector<int> vec;

    int a = 0;
    for (const auto& i : nums) {
        freq[i] = a;
        ++a;
    }

    int n = 0;
    for (int i = 0; i < nums.size(); ++i) {
        int n = target - nums[i];
        if (freq[n] && freq[n] != i) {
            return {i, freq[n]};
        }
    }

    return {};
}

int main() {
    std::vector<int> vec { 3, 1, 4, 2};
    int target = 9;

    std::vector<int> vec2 =  twoSum(vec, target);

    for(int i : vec2) {
        std::cout << i << " ";
    }

    std::cin.get();
}