PTA 1064 朋友数

题目如下：

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

输入格式：

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

输出格式：

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

输入样例：

8
123 899 51 998 27 33 36 12

输出样例：

4
3 6 9 26

解题步骤

此题比较简单，用一个函数sum计算一个int型数字各个位数的和，然后放在set中输出即可。set相对于vector的优势在于输入即排序好，不需要再次进行排序操作。但是set的输出又更加复杂，无法像vector一样直接通过索引输出，而需要通过iterator迭代器进行迭代输出。具体代码如下：

源代码

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

int sum(int num)		// 定义sum函数，求各位上的和
{
	int s=0;
	while(true)
	{
		if(num/10==0&&num%10==0)	// 循环推出条件：除以10的余数和商都为0
		{
			break;
		}
		s+=num%10;
		num/=10;
	}
	return s;
}


void fun(int a)		// 输出set
{
	cout<<" "<<a;
}

int main()
{
	int N;
	cin>>N;
	set<int> st;
	for(int i=0;i<N;i++)
	{
		int num;
		cin>>num;
		st.insert(sum(num));	// 插入set中（set可以直接自动排序）
	}
	set<int>::iterator iter=st.begin();		// 先输出第一个，控制输出格式
	cout<<st.size()<<endl;
	cout<<*iter;
	for_each(++st.begin(),st.end(),fun);
}

运行结果