合并表记录

题目描述

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

输入描述:

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

输出描述:

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

输入例子:

4
0 1
0 2
1 2
3 4

输出例子:

0 3
1 2
3 4



思路1：刚开始我想用List来存储，但是List的索引需要从0开始，若输入键值对的键（索引）不从0开始，则会出错；然后考虑使用HashMap来存储，每次输入的键值进行判断，若是键相同，则值相加，并且覆盖对应原来key的value；最后简单输入几个数字后，感觉可以，但是OJ评判错误，发现HashMap存储的数据并不一定全是有序的，少数样本有序，多数就会产生乱序，若想得到正确结果，还需进行HashMap的排序（代码在后）；然后想到使用TreeMap，TreeMap内部默认按照key升序排列，符合要求；

代码如下：

  import java.util.Map;
  import java.util.Scanner;
  import java.util.TreeMap;
  
  public class Temp {
  
      public static void main(String[] args) {
          Scanner sc = new Scanner(System.in);
          Map<Integer, Integer> m =new TreeMap<Integer, Integer>();
         while(sc.hasNextInt()){
             int n=sc.nextInt();
             for(int i=0;i<n;i++){
                 int key = sc.nextInt();
                 int value = sc.nextInt();
                 if(m.containsKey(key)){
                     m.put(key, m.get(key)+value);
                 }
                 else 
                     m.put(key, value);
             }
             for(Object key1 : m.keySet()){
                 System.out.print(key1);
                 System.out.print(" ");
                 System.out.println(m.get(key1));
                 }
         }
     }
 
 }

 

思路2：其实和思路1差不多，只是使用HashMap来存储，然后需要重新对HashMap按照key进行升序排序；这里参考别人思路，有两种方法排序

代码1如下：

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;


public class Temp {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        Map<Integer, Integer> m =new HashMap<Integer, Integer>();
        while(sc.hasNextInt()){
            int n=sc.nextInt();
            for(int i=0;i<n;i++){
                int key = sc.nextInt();
                int value = sc.nextInt();
                if(m.containsKey(key)){
                    m.put(key, m.get(key)+value);
                }
                else 
                    m.put(key, value);
            }
            
            Object[] key_arr = m.keySet().toArray();//返回一个包含HashMap中所有key的数组
            Arrays.sort(key_arr);                   //对key进行升序排列
            for(Object key1 : key_arr){
                System.out.print(key1);
                System.out.print(" ");
                System.out.println(m.get(key1));
                }
        }
    }

}

 

代码2如下：

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Scanner;


public class Temp2 {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        Map<Integer, Integer> m =new HashMap<Integer, Integer>();
        while(sc.hasNextInt()){
            int n=sc.nextInt();
            for(int i=0;i<n;i++){
                int key = sc.nextInt();
                int value = sc.nextInt();
                if(m.containsKey(key)){
                    m.put(key, m.get(key)+value);
                }
                else 
                    m.put(key, value);
            }
            
            List<Map.Entry<Integer, Integer>> li = new ArrayList<Map.Entry<Integer,Integer>>(m.entrySet());
            Collections.sort(li, new Comparator<Map.Entry<Integer,Integer>>() {

                @Override
                public int compare(Entry<Integer, Integer> o1,
                        Entry<Integer, Integer> o2) {
                    // TODO Auto-generated method stub
                    return o1.getKey().compareTo(o2.getKey());
                }
            });
            
            for(Map.Entry<Integer, Integer> map:li){
                System.out.println(map.getKey()+" "+map.getValue());
            }
            
        }
    }

}

 

参考资料：对TreeMap与HashMap重新排序

             java Map及Map.Entry详解