华为机试-8-合并表记录

合并表记录

• 题目：数据表记录包含表索引和数值，请对表索引相同的记录进行合并，即将相同索引的数值进行求和运算，输出按照key值升序进行输出。
• 思路：由于需要按照key值升序进行输出，采用TreeMap。
• 代码

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