PAT甲级_1120(Friend Numbers)

1120 友元数字

如果两个数字共享相同的数字和，那么这两个数字就被叫做“友元数字”，这个和就是它们的“友元ID”。例如：123和51是“友元数字”，因为1+2+3 = 5+1 = 6，其中6就是它们的"友元ID"。给定一些数字，你应该找到其中不同的“友元ID”的数量。

输入规范

每个输入文件包含一个测试用例。每个用例的第一行包含一个正整数N。接下来一行给出N个用空格分隔的正整数。所有的数字不超过10⁴。

输出规范

对于每个用例，在第一行打印给定整数中不同“友元ID”的个数。然后再第二行按照递增的顺序输出“友元ID”。数字之间必须精确的用一个空格隔开，并且在一行的结尾没有多余的空格。

样例输入

8
123 899 51 998 27 33 36 12

样例输出

4
3 6 9 26

解题思路

这道题我开始翻译的时候感觉有个歧义，开始以为“友元ID”至少是拥有两个数字和是这个的数字，但其实并不是。这题通俗的说只是求序列中各位数字和的所有不同情况，然后按严格递增排序输出。
可以使用set容器，这样会使代码更加简便。但是本篇文章用的是将所有数字和求出来后，进行排序，然后再去重。

AC代码

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
int sum(int n) {    //求解n的各位数字和
    int t = 0;      //记录各位数字和
    while(n) {
        t += n % 10;
        n /= 10;
    }
    return t;
}
int main() {
    int n;
    cin >> n;
    vector<int> vt;     //保存n个数字的各位数字和
    for(int i = 0; i < n; i++) {
        int x;
        cin >> x; 
        vt.push_back(sum(x));
    }
    
    sort(vt.begin(), vt.end());
    //去重
    int cnt = 1;        //记录不重复的ID个数
    for(int i = 0; i < n;) {
        int j = i+1;
        while(j < n && vt[i] == vt[j]) {
            j++;
        }
        vt[cnt] = vt[j];
        i = j;
        cnt++;
    }
    cnt--;      
    cout << cnt << endl;
    for(int i = 0; i < cnt-1; i++) {
        cout << vt[i] << ' ';
    }
    cout << vt[cnt-1] << endl;
    return 0;
}