（set get！！）PAT 乙级 1064 朋友数

如果两个整数各位数字的和是一样的，则被称为是“朋友数”，而那个公共的和就是它们的“朋友证号”。例如 123 和 51 就是朋友数，因为 1+2+3 = 5+1 = 6，而 6 就是它们的朋友证号。给定一些整数，要求你统计一下它们中有多少个不同的朋友证号。

输入格式：

输入第一行给出正整数 N。随后一行给出 N 个正整数，数字间以空格分隔。题目保证所有数字小于 104。

输出格式：

首先第一行输出给定数字中不同的朋友证号的个数；随后一行按递增顺序输出这些朋友证号，数字间隔一个空格，且行末不得有多余空格。

输入样例：

8
123 899 51 998 27 33 36 12

输出样例：

4
3 6 9 26

第一次做的时候，最后一个测试点死活都是TL，不断的优化改进之后还是TL，这时候我有以下思考：首先这道题需要判断在一个数组中有没有重复的数，如果有就不存，如果没有就存，那么我就想用find来判断

find(b,b+k,val)==b+k

其中b是数组，val是寻找的元素，k是数组的size，find函数如果没有找到那么就应该返回b的尾地址，那么这样就很好理解了，但是很显然，TL，第一次代码如下（冗余切复杂）

#include<bits/stdc++.h>
using namespace std;
int Sum(int a)
{
    int suma = 0;
    while (a)
    {
        suma += a % 10;
        a /= 10;
    }
    return suma;
}
bool friends(int a, int b)
{
    int suma = Sum(a);
    int sumb = Sum(b);

    return suma == sumb;
}
int main()
{
    int n=;
   scanf("%d", &n);
    int s[n];
    int i = 0;

    while (i < n)
    {
        scanf("%d", &s[i++]);
    }
    int count = 0;
    int k = 0;
    int b[1000];
    memset(b, 0, sizeof(b));
    for (int i = 0; i < n; i++)
        for (int j = i + 1; j < n; j++)
        {
            if (friends(s[i], s[j]))
            {
                if (count == 0)
                {
                    b[k++] = Sum(s[j]);
                    count++;
                }

                else if (find(b, b + k, Sum(s[j])) == b + k) {
                    count++;
                    b[k++] = Sum(s[j]);
                }
            }
        }
    sort(b, b + k);
    printf("%d ", count);
    for (int i = 0; i < k; i++)
    {
        if (i != k - 1)
            printf("%d ", b[i]);
        else printf("%d", b[i]);
    }
}

因为一直优化一直TL，我不知道该怎么办，这时候我想到了容器set！没错 set是根据二叉树的原理，存储没有重复的数据并且排序，我想这不正好完美符合题意，所以我选择set，set的用法让我们看下

set<数据类型>[名字];

这样就能定义一个容器

set<int>st;
st.insert(a);//将a放入set
set容器：
所有元素在插入时自动被排序，属于关联式容器，底层结构为二叉树
set与multiset区别：set不允许有重复元素，而multiset可以
构造和赋值：
set<int>s1
=赋值
插入：只有insert方式
set.insert(elem)注意：若要是插入后元素是降序排列，则要使用仿函数作为排列规则
具体就是写一个函数，让他降序，类似于sort
大小和交换：

set.size()
set.empty()
set.swap(l)

插入和删除：

set.insert(elem)
set.clear()
set.erase(pos)
set.erase(betg,end)
set.erase(elem)

所以很简单咯

代码如下

但是遍历的时候要用迭代器，比如set<int>::interator it=st.begin();//类似于创建了一个it指针指向set的首地址

然后

for(;it!=st.end();it++)

//visit

#include<bits/stdc++.h>
using namespace std;
int friends(int a)
{
    int suma=0;
    while(a)
    {
        suma+=a%10;
        a/=10;
    }
    return suma;
}
int main()
{
    set<int>st;
    int n;
    cin>>n;
    while(n--)
    {
        int x;
        cin>>x;
        st.insert(friends(x));

    }
    cout<<st.size()<<endl;
    int flag=0;
    for(set<int>::iterator it=st.begin();it!=st.end();it++)
    {
        if(!flag) {
            cout << *it ;
            flag=1;
        }
        else cout<<" "<<*it;
    }

}