PAT (Basic Level) Practice 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

 ps：打个草稿

#define _CRT_SECURE_NO_WARNINHS
#define _CRT_SECURE_NO_DEPRECATE

#include<iostream>
#include<stdio.h>
#include<string>
#include<algorithm>
#include<numeric>
#include<functional>
#include<vector>
#include<stack>
#include<math.h>
#include<stdlib.h>
#include<unordered_map>
#include<unordered_set>

using namespace std;


///1064 朋友数
int toto(int a)
{
	if (a >= 1000)
	{
		return (a / 1000 + (a / 100) % 10 + (a / 10) % 10 + a % 10 );
	}
	else if (a >= 100 && a < 1000)
	{
		return (a / 100 + (a / 10)%10 + a % 10);
	}
	else if (a>=10 && a < 100)
	{
		return (a / 10 + a % 10);
	}
	else if (a>0 && a < 10)
	{
		return a;
	}
}
int main()
{
	int n, i, j;
	cin >> n;

	int num[10001];
	for (i = 0; i < n; i++)
	{
		cin >> num[i];
	}

	vector<int> total;
	for (i = 0; i < n; i++)
	{
		for (j = i+1; j < n; j++)
		{
			if (toto(num[i]) == toto(num[j]))
			{
				total.push_back(toto(num[i]));
			}
		}
	}
	cout << total.size() << endl;
	for (i = 0; i < total.size(); i++)
	{
		if (i != 0)
			cout << " ";
		cout << total[i];
	}


	return 0;
}

ps：然后突然想起set的应用，set是有序的，不重复的，非常适合用在这里

#define _CRT_SECURE_NO_WARNINHS
#define _CRT_SECURE_NO_DEPRECATE

#include<iostream>
#include<stdio.h>
#include<string>
#include<algorithm>
#include<set>

using namespace std;


///1064 朋友数
int toto(int a)
{
	if (a >= 1000)
	{
		return (a / 1000 + (a / 100) % 10 + (a / 10) % 10 + a % 10 );
	}
	else if (a >= 100 && a < 1000)
	{
		return (a / 100 + (a / 10)%10 + a % 10);
	}
	else if (a>=10 && a < 100)
	{
		return (a / 10 + a % 10);
	}
	else if (a>0 && a < 10)
	{
		return a;
	}

	/*while (a != 0)
	{
		a += a % 10;
		a / 10;
	}*/
}
int main()
{
	int n, i, j;
	cin >> n;

	int num[10001];
	for (i = 0; i < n; i++)
	{
		cin >> num[i];
	}
	
	set<int> total;

	for (i = 0; i < n; i++)
	{
		total.insert(toto(num[i]));
	}
	cout << total.size() << endl;
	for (auto it = total.begin(); it != total.end(); it++)
	{
		if (it != total.begin())
			cout << " ";
		cout << *it;
	}

	return 0;
}