PAT 乙级 1064 朋友数 (20 分)

myriadddddd

于 2022-02-28 10:47:38 发布

阅读量74

点赞数

文章标签：算法 c++ 数据结构 PAT

本文链接：https://blog.youkuaiyun.com/qq_42952157/article/details/123177487

版权

本文介绍了一种算法，用于统计一组整数中不同朋友证号的数量，并展示了如何通过set数据结构和getFriendNum函数来解决这个问题。通过输入样例8个整数的实例，演示了如何找出并输出它们的不同朋友证号：36926。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

输入格式：

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

输出格式：

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

输入样例：

8
123 899 51 998 27 33 36 1232

输出样例：

4
3 6 9 26

分析：

set中元素的排序默认就是升序且元素互不相同，刚好满足题目的要求；剩下的我们只要写一个将数字转化成朋友数的函数即可。

代码：

#include <iostream>
#include <set>

using namespace std;

// 转化函数
int getFriendNum(int val) {
    int temp = 0;
    while (val != 0) {
        temp += (val % 10);
        val /= 10;
    }

    return temp;
}

int main() {
    int N;
    scanf("%d", &N);
    set<int> st;

    for (int i = 0; i < N; i++) {
        int temp;
        scanf("%d", &temp);
        st.insert(getFriendNum(temp));  // 插入 st 中
    }

    cout << st.size() << endl;

    // 遍历 st
    for (auto it = st.begin(); it != st.end(); it++) {
        if (it != st.begin())
            printf(" ");
        cout << *it;
    }
    
    return 0;
}