【华为OJ】【008-合并表记录】

【华为OJ】【算法总篇章】

【华为OJ】【008-合并表记录】

【工程下载】

题目描述

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

输入描述

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

输出描述

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

输入例子

4
0 1
0 2
1 2
3 4

输出例子

0 3
1 2
3 4

算法实现

import java.util.Scanner;
import java.util.SortedMap;
import java.util.TreeMap;

/** * Author: 王俊超 * Date: 2015/12/21 16:33 * All Rights Reserved !!! */
public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);

        while (scanner.hasNext()) {
            int pares = Integer.parseInt(scanner.nextLine());
            SortedMap<Integer, Integer> map = new TreeMap<>();
            for (int i = 0; i < pares; i++) {
                String[] nums = scanner.nextLine().split("\\s+");
                addPare(map, nums);
            }

            System.out.print(mapToString(map));
        }

        scanner.close();
    }

    private static void addPare(SortedMap<Integer, Integer> map, String[] nums) {

        int key = Integer.parseInt(nums[0]);
        int val = Integer.parseInt(nums[1]);

        if (map.containsKey(key)) {
            map.put(key, map.get(key) + val);
        } else {
            map.put(key, val);
        }
    }

    private static String mapToString(SortedMap<?, ?> map) {

        StringBuilder builder = new StringBuilder();

        for (SortedMap.Entry<?, ?> e : map.entrySet()) {
            builder.append(e.getKey()).append(" ").append(e.getValue()).append("\n");
        }

        return builder.toString();
    }
}