這裏先看一道題目:
題目描述
數據表記錄包含表索引和數值,請對錶索引相同的記錄進行合併,即將相同索引的數值進行求和運算,輸出按照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;
}