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

38

39

40

41

42

43

44

#include <vector>

#include <map>

/**

*

*給定一個整數陣列和一個目標值，找出陣列中和為目標值的兩個數。

*你可以假設每個輸入只對應一種答案，且同樣的元素不能被重複利用。

*case:

*給定 nums = [2, 7, 11, 15], target = 9

*因為 nums[0] + nums[1] = 2 + 7 = 9

*所以返回 [0, 1]

*/

using namespace std;

class Solution {

public:

// 兩兩對比的方式，時間複雜度為O(n2)

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

vector<int> ret;

// 第一種

/*for (int i = 0; i < nums.size(); i++) {

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

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

ret.push_back(i);

ret.push_back(j);

return ret;

}

}

}*/

// 第二種：資料插入到hashmap裡，然後通過target - nums[i]來確定，時間複雜度為O(2n)

map<int, int> m;

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

m[nums[i]]  = i;

}

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

if (m.count(target - nums[i]) > 0 && i != m[target - nums[i]]) {

ret.push_back(i);

ret.push_back(m[target - nums[i]]);

return ret;

}

}

return ret;

}

};