PAT甲组1120 Friend Numbers思路解析和代码

A1120

题目链接

个人思路

水题，使用STL中的set容器效果最佳，去重+排序
数字各位求和，输出不同和的个数并排序
注意行末空格问题，用计数器和set.size()比较即可

个人思路代码

#include <bits/stdc++.h>
using namespace std;
int N;
set<int> friendid;
int getid(int n)
{
	int ans = 0;
	while(n)
	{
		int ret = n % 10;
		ans += ret;
		n /= 10;
	}
	return ans;
}
int main(int argc, char *argv[]) {
	scanf("%d", &N);
	for(int i = 0; i < N; ++i)
	{
		int num;
		scanf("%d", &num);
		int id = getid(num);
		friendid.insert(id);
	}
	printf("%d\n", friendid.size());
	int cnt = 0;
	for(set<int>::iterator it = friendid.begin(); it != friendid.end(); ++it)
	{
		printf("%d", *it);
		cnt++;
		if(cnt < friendid.size())
			printf(" ");
	}
	return 0;
}