PTA 乙级 1064 朋友数 (20分) C++

 

 思路：对于每个输入的数每位相加后的数（朋友数）存入一个vector中，利用sort对朋友数进行升序排序，对所有朋友数进行遍历，统计所有不重复的的朋友数的总数，输出总数后，再次对所有朋友数进行遍历，输出不重复的朋友数（这个方法是我第一次想出来的方法，有些繁琐，可以尝试用集合的思路）

#include<iostream>
#include<vector>
#include<algorithm>

using namespace std;

int main() {
    int n = 0;
    int num = 0;
    int flag = 0;
    int count = 0;
    cin >> n;
    vector<int> sum(n);
    for (int i = 0; i < n; ++i) {
        cin >> num;
        int tmp = -1;
        while (tmp != 0) {            //每位相加
            tmp = num % 10;
            num /= 10;
            sum[i] += tmp;
        }
    }
    sort(sum.begin(), sum.end());    //递增排序
    int tmp1 = 0;
    for (int i = 0; i < n; ++i) {    //计算总共有多少个朋友数
        if (sum[i] != tmp1) { count++; tmp1 = sum[i]; }
    }
    cout << count << endl;
    int tmp2 = 0;
    for (int i = 0; i < n; ++i) {
        if (sum[i] != tmp2) {        //保证不会输出重复的
            if (flag) cout << ' ';
            cout << sum[i];
            tmp2 = sum[i];
            flag = 1;
        }
    }
    return 0;
}