PAT1120 Friend Numbers 找朋友（STL-SET）

如果两个整数各位数字的和是一样的，则被称为是“朋友数”，而那个公共的和就是它们的“朋友证号”。

例如 123和 51就是朋友数，因为 1+2+3=5+1=6，而 6 就是它们的朋友证号。

给定一些整数，要求你统计一下它们中有多少个不同的朋友证号。

 

思路：输入时用字符串加起来方便，用set可以直接去重。

#include<bits/stdc++.h>
#define rep(i,a,n) for(int i=a;i<n;i++)
#define INF 0x3f3f3f;
using namespace std;
const int N=1e5+233;
int n;
set<int>s;
int gao(string str){
    int res=0;
    rep(i,0,str.length()) res+=str[i]-'0';
    return res;
}
int main(){
    cin>>n;
    string str;
    rep(i,0,n){
        cin>>str;
        s.insert(gao(str));
    }
    cout<<s.size()<<endl;
    int flag=0;
    for(auto u:s){
        if(flag) cout<<" ";
        cout<<u;
        flag++;
    }
    cout<<endl;
}