华为OJ——合并表记录

最新推荐文章于 2022-05-18 18:45:18 发布

大多山

最新推荐文章于 2022-05-18 18:45:18 发布

阅读量302

点赞数

分类专栏：华为OJ系列文章标签：合并索引 java算法

本文链接：https://blog.csdn.net/qq_36295256/article/details/74853066

版权

华为OJ系列专栏收录该内容

61 篇文章 1 订阅

订阅专栏

该博客介绍了一道关于合并表记录的算法题，要求对相同索引的数值进行求和，最后按升序输出键值对。给出了利用Java中TreeMap的解决方案，利用其自然排序和键的唯一性特性。

摘要由CSDN通过智能技术生成

题目描述

　　数据表记录包含表索引和数值，请对表索引相同的记录进行合并，即将相同索引的数值进行求和运算，输出按照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();
    }
}