PAT(甲级) 1120 Friend Numbers (20point(s)) set的使用

题目

题目链接

思路

依次处理每个数字，把每位数字求和后放入set容器中（set容器会自动排序）

代码

#include <iostream>
#include <vector>
#include <cstring>
#include <algorithm>
#include <stack>
#include <set>
#include <queue>
#include <malloc.h>
#include <math.h>

using namespace std;
const int maxn = 1010;

set<int> S;

int main()
{
     int n, t, sum = 0;
     scanf("%d", &n);
     for(int i = 0; i < n; i ++){
          scanf("%d", &t);
          sum = 0;
          while(t > 0){
              sum += t % 10;
              t = t / 10;
          }
          S.insert(sum);
     }
     printf("%d\n", (int)S.size());
     bool flag = false;
     for(auto it = S.begin(); it != S.end(); it ++){
          if(!flag) flag = !flag;
          else printf(" ");
          printf("%d", *it);
     }
     printf("\n");
     system("pause");
     return 0;
}