华为OJ——合并表记录

题目描述

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

输入描述:

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

输出描述:

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

输入例子:

　　4
　　0 1
　　0 2
　　1 2
　　3 4

输出例子:

　　0 3
　　1 2
　　3 4

代码实现

思路：根据TreeMap默认自然排序，且键不可重复

package cn.c_shuang.demo9;import java.util.Iterator;import java.util.Map;import java.util.Map.Entry;import java.util.Scanner;import java.util.Set;import java.util.TreeMap;/** * 数据表记录包含表索引和数值，请对表索引相同的记录进行合并，即将相同索引的数值进行求和运算，输出按照key值升序进行输出。 * @author Cshuang * */public class Main {    public static void main(String[] args) {        Scanner in=new Scanner(System.in);        while(in.hasNext()){            int cnt=in.nextInt();            Map<Integer,Integer> map=new TreeMap<Integer,Integer>();            for (int i = 0; i < cnt; i++) {                int key=in.nextInt();                int value=in.nextInt();                if(map.containsKey(key)){                    int newValue=map.get(key)+value;                    map.put(key, newValue);                } else{                    map.put(key, value);                }            }            Set<Map.Entry<Integer, Integer>> entrySet =map.entrySet();            Iterator<Entry<Integer, Integer>> it=entrySet.iterator();            while(it.hasNext()){                Map.Entry<Integer,Integer> me =it.next();                System.out.println(me.getKey()+" "+me.getValue());            }        }        in.close();    }}

改进：省略了迭代器

import java.util.Map;import java.util.Scanner;import java.util.TreeMap;public class Main {    public static void main(String[] args) {        Scanner in=new Scanner(System.in);        while(in.hasNext()){            int cnt=in.nextInt();            Map<Integer,Integer> map=new TreeMap<Integer,Integer>();            for (int i = 0; i < cnt; i++) {                int key=in.nextInt();                int value=in.nextInt();                if(map.containsKey(key)){                    int newValue=map.get(key)+value;                    map.put(key, newValue);                } else{                    map.put(key, value);                }            }            for (Integer k : map.keySet()) {                System.out.println(k+" "+map.get(k));            }        }        in.close();    }}