Map操作之merge

這裏先看一道題目:
題目描述
數據表記錄包含表索引和數值,請對錶索引相同的記錄進行合併,即將相同索引的數值進行求和運算,輸出按照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;
    }
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章