第十四天PAT-A1120 Friend Numbers字符串简单模拟

A1120

Description：

如果两个数每位的和相同，则称为一对朋友数，和为其朋友ID；现给出一系列数，求这些数中不同的朋友ID数目并分别输出这些朋友ID

思路：

• 模拟题，把每个数字的朋友ID计算出来放到一个set里；
• 担心超时所以没用string存数；

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<math.h>
#include<string.h>
#include<algorithm>
#include<map>
#include<set>
#include<vector>
#include<queue>
using namespace std;
char num[10];
set<int>ans;
int n;
int main()
{
#ifdef ONLINE_JUDGE
#else
    freopen("1.txt", "r", stdin);
#endif // ONLINE_JUDGE
    scanf("%d", &n);
    for(int i = 0; i < n; i++){
        scanf("%s", &num);
        int cnt = 0;
        for(int j = 0; j < strlen(num); j++){
            cnt += (num[j] - '0');
        }
        ans.insert(cnt);
    }
    printf("%d\n", ans.size());
    for(auto it = ans.begin(); it != ans.end(); it++){
        if(it != ans.begin()) printf(" ");
        printf("%d", *it);
    }
    printf("\n");
    return 0;
}