个人练习-PAT甲级-1120 Friend Numbers

最新推荐文章于 2025-12-06 19:00:36 发布

原创最新推荐文章于 2025-12-06 19:00:36 发布 · 180 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #算法

这篇博客介绍了如何使用C++解决一个编程问题，即计算给定N个十进制数中，具有相同位和作为ID的朋友数，并按ID排序输出。核心思路是遍历输入，用哈希表记录ID并计数。

题目链接https://pintia.cn/problem-sets/994805342720868352/problems/994805352925609984

题目大意：给出一个十进制数，其所有位的和是它的ID，ID相同的数叫朋友数。给出N个数字，求有几个不同的ID，并将这些ID排序输出。

思路：简单题，正常算即可。

完整代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <vector>
#include <algorithm>
#include <queue>
#include <map>
#include <set>


using namespace std;


bool known[37];
vector<int> arr;

int calId(int num) {
    int sum = 0;
    while (num != 0) {
        sum += num % 10;
        num /= 10;
    }
    return sum;
}


int main() {
    int N;
    scanf("%d", &N);
    for (int i = 0; i < N; i++) {
        int num, id;
        scanf("%d", &num);
        id = calId(num);
        if (!known[id]) {
            known[id] = true;
            arr.push_back(id);
        }
    }

    sort(arr.begin(), arr.end());

    printf("%d\n%d", arr.size(), arr[0]);
    for (int i = 1; i < arr.size(); i++)
        printf(" %d", arr[i]);

    return 0;
}