[LeetCode] 170、两数之和 III - 数据结构设计

题目描述

设计并实现一个 TwoSum 的类，使该类需要支持 add 和 find 的操作。

• add 操作 - 对内部数据结构增加一个数。

• find 操作 - 寻找内部数据结构中是否存在一对整数，使得两数之和与给定的数相等。

示例：

add(1); add(3); add(5);
find(4) -> true
find(7) -> false

解题思路

几数之和问题，可考虑用哈希表实现。

参考代码

class TwoSum {
public:
    /** Initialize your data structure here. */
    TwoSum() {
        
    }
    
    /** Add the number to an internal data structure.. */
    void add(int number) {
        umap[number]++;
    }
    
    /** Find if there exists any pair of numbers which sum is equal to the value. */
    bool find(int value) {
        for(auto elem: umap){   
            if(elem.first == value - elem.first && umap[value - elem.first] >= 2)  // 注意特殊情况
                return true;
            else if(elem.first != value - elem.first && umap.count(value - elem.first) > 0)
                return true;
        }
        return false;
    }
    
private:
    unordered_map<int, int> umap;
};

/**
 * Your TwoSum object will be instantiated and called as such:
 * TwoSum* obj = new TwoSum();
 * obj->add(number);
 * bool param_2 = obj->find(value);
 */