【刷题】华为笔试面试机考 [HJ8] - 合并表记录

题目地址

• 点击跳转

题目描述

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

输入描述:

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

输出描述:

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

示例1

输入

4
0 1
0 2
1 2
3 4

输出

0 3
1 2
3 4

解题思路

接收，分组（同时计算），排序，输出。

代码

• Python 3

k = int(input())

rs = {}
while k > 0:
    k -= 1
    
    rk, rv = map(int, input().split())
    
    if rk in rs:
        rs[rk] += rv
        continue
    rs[rk] = rv
    
rsk = sorted(rs)
for k in rsk:
    print("{} {}".format(k, rs[k]))