华为机试 牛客网 HJ8 合并表记录

描述

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

输入描述：

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

输出描述：

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

示例1

输入：
4
0 1
0 2
1 2
3 4

输出：
0 3
1 2
3 4

题解

import java.util.*;
public class Main{
    
    public static void main(String[] args){
        
        Scanner sc = new Scanner(System.in);
        
        int num = sc.nextInt();
        TreeMap<Integer,Integer> treemap = new TreeMap<>();
        
        for(int i=0;i<num;i++){
            
            int key = sc.nextInt();
            int value = sc.nextInt();
            if(treemap.containsKey(key))
                treemap.put(key,treemap.get(key)+value);
            else
                treemap.put(key,value);
            
            
        }
        
        for(Map.Entry<Integer,Integer> entry:treemap.entrySet())
            System.out.println(entry.getKey()+" "+entry.getValue());
        
    }
    
    
}

反思

要会treemap的使用与特点，以及他的遍历