华为机试：合并表记录

题目描述

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

输入描述:

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

输出描述:

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

示例1

输入

4
0 1
0 2
1 2
3 4

输出

0 3
1 2
3 4

 

Java：将数据存储到二维数组中，然后在Arrays.sort()中重写了Comparator，让其可以直接对二维数组利用某一列进行关键字排序，排序后关键字相同的行均相邻，将相邻的行的值value相加存储到另一个数组中即可。这个方法并没有用库函数。代码长度较长，做题时不宜实现，但时间复杂度、空间复杂度上来讲两者一样，甚至有些小优势。

import java.util.Arrays;
import java.util.Comparator;
import java.util.Scanner;

public class Main {
    
    public static void main(String[] args) {
        Scanner sc=new Scanner(System.in);
        while(sc.hasNext()){
            int num = sc.nextInt();
            int[][] nums = new int [num][2];
            for(int i = 0;i<num;i++){
                nums[i][0] = sc.nextInt();
                nums[i][1] = sc.nextInt();
            }
            
            Arrays.sort(nums, new Comparator<int[]>(){

                @Override
                public int compare(int[] o1, int[] o2) {
                    // TODO Auto-generated method stub
                    if(o1[0]>o2[0]){
                        return 1;
                    }
                    else{
                        return -1;
                    }
                }});
            
            int count=1;
            int temp=nums[0][0];
            for(int i = 1; i<nums.length;i++){
                if(temp==nums[i][0]){
                }
                else{
                    count++;
                    temp=nums[i][0];
                }
            }
            
            int[][] end = new int[count][2];
            end[0]=nums[0];
            int j=0;
            for(int i = 1; i<nums.length;i++){
                if(end[j][0]==nums[i][0]){
                    end[j][1]+=nums[i][1];
                }
                else{
                    j++;
                    end[j] = nums[i];
                }
            }

            for(int i = 0; i < end.length; i++){
                System.out.println(end[i][0]+" "+end[i][1]);
            }
            
        }
        sc.close();
    }
    
}

Java：利用map中的Treemap数据结构，Treemap在构建的过程中直接是关键字有序的，其内部实现为排序二叉树（1.8改为红黑树）。*注：抄袭别人的，不知道这个数据结构，当然红黑树也不清楚。

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

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

C++：