每日编程292期——朋友数

解决方法

（1）算法的基本思想：

计算输入数字的各位数字的和，将其添加在set中即可。
需要注意的是：最后一个数字后面不得有空格

（2）代码实现

#include<iostream>
#include<iterator>
#include<set>
#include<vector>
using namespace std;

struct number
{
    int n;
    int sum;
};

int main(){
    int n;
    cin >> n;
    vector<number> num(n);
    set<int> s;
    for(int i = 0; i < n ; i++){
        cin >> num[i].n;
        while(num[i].n != 0){
            num[i].sum += num[i].n % 10;
            num[i].n /= 10;
        }
        s.insert(num[i].sum);
    }
    cout << s.size() <<endl;//set去重且有序
    int max = *(--s.end());
    s.erase(--s.end());//为了最后没有空格，把最后一个单独输出
    copy(s.begin(), s.end(), ostream_iterator<int>(cout, " "));
    cout << max;
    return 0;
}

set是关联容器的一种，具体实现采用了红黑树的平衡二叉树的数据结构。由于采用了红黑树(Red-Black Tree)的平衡二叉检索树的数据结构，在插入元素时, 它会自动调整二叉树的排列，把该元素放到适当的位置，以确保每个子树根节点的键值大于左子树的所有节点的键值，而小于右子树所有节点的键值；另外，还得确保根节点左子树的高度与右子树的高度相等，这样，二叉树的高度最小，从而检索速度最快。要注意的是，它不会重复插入相同键值的元素，而采取忽略处理。

平衡二叉检索树的检索使用中序遍历算法，检索效率高于vector, deque和list等容器。另外，采用中序遍历算法可将键值由小到大遍历出来，所以，可以理解为平衡二叉检索树在插入元素时，就会自动将元素按键值由小到大的顺序排列。

对于set容器中的键值，不可直接去修改。因为如果把容器中的一个键值修改了，set容器会根据新的键值旋转子树，以保持新的平衡，这样，修改的键值很可能就不在原先那个位置上了，换句话来说，构造set集合的主要目的就是为了快速检索。