PAT乙级真题 1064朋友数(20)

#include<iostream>
#include <set>
#include <vector>
using namespace std;
int part(int num)
{
int sum = 0;
while (num != 0) {
sum += num % 10;
num /= 10;
}
return sum;
}
int main()
{
int N;
cin >> N;
vector<int> A(N);
for (int i = 0; i < N; i++)
cin >> A[i];
set<int> s;
for (int i = 0; i < N; i++)
{
s.insert(part(A[i]));
}
cout << s.size()<<endl;
for (set<int>::iterator it = s.begin(); it != s.end(); it++)
{
if (it!=s.begin())
{
cout << " ";
}
cout << *it ;
}
return 0;
}