170. Two Sum III - Data structure design【easy】
Design and implement a TwoSum class. It should support the following operations:add and find.
add - Add the number to an internal data structure.
find - Find if there exists any pair of numbers which sum is equal to the value.
For example,
add(1); add(3); add(5);
find(4) -> true
find(7) -> false
这是一道收费题目,参考了grandyang的博客,感谢他!
解法一:
1 class TwoSum { 2 public: 3 void add(int number) { 4 ++m[number]; 5 } 6 bool find(int value) { 7 for (auto a : m) { 8 int t = value - a.first; 9 if ((t != a.first && m.count(t)) || (t == a.first && a.second > 1)) { 10 return true; 11 } 12 } 13 return false; 14 } 15 private: 16 unordered_map<int, int> m; 17 };
哈希表的解法,我们把每个数字和其出现的次数建立映射,然后我们遍历哈希表,对于每个值,我们先求出此值和目标值之间的差值t,然后我们需要分两种情况来看,如果当前值不等于差值t,那么只要哈希表中有差值t就返回True,或者是当差值t等于当前值时,如果此时哈希表的映射次数大于1,则表示哈希表中还有另一个和当前值相等的数字,二者相加就是目标值
解法二:
1 class TwoSum { 2 public: 3 void add(int number) { 4 s.insert(number); 5 } 6 bool find(int value) { 7 for (auto a : s) { 8 int cnt = a == value - a ? 1 : 0; 9 if (s.count(value - a) > cnt) { 10 return true; 11 } 12 } 13 return false; 14 } 15 private: 16 unordered_multiset<int> s; 17 };
另一种解法不用哈希表,而是unordered_multiset来做,但是原理和上面一样
参考自:http://www.cnblogs.com/grandyang/p/5184143.html