这里先看一道题目:
题目描述
数据表记录包含表索引和数值,请对表索引相同的记录进行合并,即将相同索引的数值进行求和运算,输出按照key值升序进行输出。
输入描述:
先输入键值对的个数
然后输入成对的index和value值,以空格隔开
输出描述:
输出合并后的键值对(多行)
输入
4
0 1
0 2
1 2
3 4
输出
0 3
1 2
3 4
要你做,怎么做呢?怎么才能最简洁呢?
方式1:写一个类,v1为key,v2为value
然后放在list里面;
然后对list进行流操作,根据v1分组,同时求和v2;
最后遍历输出;
方式2:利用Map的Merge操作,可以自定义操作;
简写如下:
public static void main(String[] args) {
Map<Integer,Integer> map=new TreeMap<>();
// map.merge(1, 2, (old,curr)->(old+curr));
// map.merge(1, 3, (old,curr)->(old+curr));
//简写
map.merge(1, 2, Integer::sum);
map.merge(1, 3, Integer::sum);
map.merge(1, 4, Integer::sum);
map.merge(2, 4, Integer::sum);
map.forEach((k,v)->System.out.println(k+" "+v));
}
源码如下:
default V merge(K key, V value,
BiFunction<? super V, ? super V, ? extends V> remappingFunction) {
Objects.requireNonNull(remappingFunction);
Objects.requireNonNull(value);
V oldValue = get(key);
V newValue = (oldValue == null) ? value :
remappingFunction.apply(oldValue, value);
if(newValue == null) {
remove(key);
} else {
put(key, newValue);
}
return newValue;
}