题目:
设计一个类,包含如下两个成员函数:
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
}