170. Two Sum III - Data structure design【easy】

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

 

转载于:https://www.cnblogs.com/abc-begin/p/7715952.html