题目:
设计一个类,包含如下两个成员函数:
Save(int input) 插入一个整数到一个整数集合里,整数集中可以存在相同值。
Test(int target) 检验是否存在两个数和为输入值。如果存在这两个数,则返回true,否则返回false。
书中给出的解析:这题相比上一题,只要把<值,下标>改成<值,个数>,第一次遍历将数组存到哈希表里面;第二次遍历数组,判断这个和减去当前数组的值的差值,如果差值在哈希表中存在,满足下面两个条件之一则返回true,(1)当前值不等于差值(2)当前值等于差值,但是在哈希表中该键值对应的值大于等于2。
代码实现:
type TwoSum struct { hm []int } func (ts *TwoSum) Save(input int) { ts.hm = append(ts.hm, input) } func (ts *TwoSum) Test(target int) bool { m, length := make(map[int]int), len(ts.hm) for i := 0;i < length;i += 1 { m[ts.hm[i]] += 1 } for i := 0;i < length;i += 1 { v := m[target-ts.hm[i]] if (2 * ts.hm[i] != target && v != 0) || (2 * ts.hm[i] == target && v > 1) { fmt.Println(ts.hm[i], target - ts.hm[i]) return true } } return false }