1064. 朋友数(20)

发表于 | 分类于 ，

题目描述

如果两个整数各位数字的和是一样的，则被称为是“朋友数”，而那个公共的和就是它们的“朋友证号”。例如123和51就是朋友数，因为1+2+3 = 5+1 = 6，而6就是它们的朋友证号。给定一些整数，要求你统计一下它们中有多少个不同的朋友证号。注意：我们默认一个整数自己是自己的朋友。

输入格式：

输入第一行给出正整数N。随后一行给出N个正整数，数字间以空格分隔。题目保证所有数字小于104。

输出格式：

首先第一行输出给定数字中不同的朋友证号的个数；随后一行按递增顺序输出这些朋友证号，数字间隔一个空格，且行末不得有多余空格。

输入样例：

8
123 899 51 998 27 33 36 12

输出样例：

4
3 6 9 26

提交代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
#include <iostream>
#include <set>
using namespace std;
int main()
{
    int N, num, count;
    set<int> friend_id;

    cin >> N;
    while (N--) {
        cin >> num;
        count = 0;
        while (num) {
            count += num % 10;
            num /= 10;
        }
        friend_id.insert(count);
    }

    cout << friend_id.size() << endl;
    cout << *friend_id.begin();
    friend_id.erase(*friend_id.begin());
    for (auto it = friend_id.begin(); it != friend_id.end(); it++) {
        cout << " " << *it;
    }
    cout << endl;

    return 0;
}

个人思考

这里用 std::set 容器来做会比较好，但因为数据量也不大，所以一开始就直接开个数组来做了。

更新历史

  • 2018/04/21：更新为使用了 std::set 的代码，STL 容器是个好东西。