PAT-B-1064-朋友数

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

输入格式：

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

输出格式：

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

输入样例：

1 2	8 123 899 51 998 27 33 36 12

输出样例：

1 2	4 3 6 9 26

Code

#include <iostream>

using namespace std;
int record[90000] = { 0 };
int getFriendN(int num) {
	int res = 0;
	do {
		int temp = num % 10;
		res += temp;
		num /= 10;
	} while (num != 0);
	return res;
}
int main() {

	int n, num, count = 0;
	scanf("%d", &n);

	for (int i = 0;i < n;i++) {
		scanf("%d", &num);
		int index = getFriendN(num);
		//单独也算朋友数
		if (record[index] == 0) 
			count++;
		record[index]++;
	}
	printf("%d\n", count);

	for (int i = 0;i < 90000;i++) {
		if (count == 0)
			break;
		if (record[i]!=0) {
			printf("%d", i);
			count--;
			if (count != 0)
				printf(" ");
		}
			
		
	}

	return 0;
}