牛客网 - 在线编程 - 华为机试 - 合并表记录

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

输入描述:

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

输出描述:

输出合并后的键值对(多行)

示例1
输入

4
0 1
0 2
1 2
3 4

输出

0 3
1 2
3 4

C++:

参考map详解

#include <iostream>
#include <map>
using namespace std;

int main()
{

    int n;
    while (cin >> n)
    {
        map<int, int> mymap;
        while (n--)
        {
            int a, b;
            cin >> a; cin >> b;
            auto it = mymap.find(a);
            if (it != mymap.end())
                mymap[a] += b;
            else
                mymap[a] = b;
        }
        for (auto it = mymap.begin(); it != mymap.end(); ++it)
        {
            cout << it->first << " " << it->second << endl;
        }
    }
    return 0;
}

Python:

n = int(input())
dic = {}

for i in range(n):
    key, val = map(int, input().split())
    if key in dic:
        dic[key] += val
    else:
        dic[key] = val
for i,j in sorted(dic.items()):
    print(i,j)

猜你喜欢

转载自blog.csdn.net/qq_39735236/article/details/82689228