【牛客网机试题】合并表记录

题目描述

数据表记录包含表索引和数值（int范围的整数），请对表索引相同的记录进行合并，即将相同索引的数值进行求和运算，输出按照key值升序进行输出。

输入描述:

先输入键值对的个数
然后输入成对的index和value值，以空格隔开

输出描述:

输出合并后的键值对（多行）

示例1

输入

复制

4
0 1
0 2
1 2
3 4

输出

复制

0 3
1 2
3 4


import java.util.Map;
import java.util.Scanner;
import java.util.Set;
import java.util.TreeMap;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int size = scanner.nextInt();
        Map<Integer, Integer> keyValue = new TreeMap<>();
        for (int i = 0; i < size; i++) {
            int key = scanner.nextInt();
            int value = scanner.nextInt();
            if (keyValue.containsKey(key)) {
                keyValue.put(key, keyValue.get(key) + value);
                continue;
            }
            keyValue.put(key, value);

        }
        Set<Map.Entry<Integer, Integer>> entries = keyValue.entrySet();
        for (Map.Entry<Integer, Integer> string : entries) {
            System.out.println(string.getKey() + " " + string.getValue());
        }
    }
}