【练习】PAT 乙 1064. 朋友数(20)

题目

1064. 朋友数(20)

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

输入格式：

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

输出格式：

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

输入样例：

8

123 899 51 998 27 33 36 12

输出样例：

4

3 6 9 26

来源：PAT 乙 1064. 朋友数(20)

---

思路（注意事项）

用哈希表存放朋友证号，会自动排列大小，然后依次输出即可。

---

纯代码

#include<bits/stdc++.h>
using namespace std;
int main(){
	int n, num = 0;
	cin >> n;
	
	map<int,int> mp;
	for (int i = 0; i < n ; i ++)
	{
		int t;
		cin >> t;
		
		string s = to_string(t);
		int sum = 0;
		for (int j = 0; j < s.size(); j ++) sum += s[j] - '0';
		
		if (mp.find(sum) == mp.end()) 
		{
			mp[sum] = 1;
			num ++;
		}
	}
	cout << num << endl;
	int flag = 1;
	for (auto i : mp) 
	{
		if (flag == 0) cout << " ";
		else flag = 0;
		cout << i.first;
	}
	cout << endl;
	return 0;
} 

题解（加注释）

#include<bits/stdc++.h>

using namespace std;

const int N = 10010;  // 定义一个常量 N，表示数组的最大长度
int a[N];  // 定义一个数组 a，未使用

int main(){
    int n, num = 0;  // 定义变量 n 表示输入的数字个数，num 表示不同数字和的个数
    cin >> n;  // 输入数字个数 n
    
    map<int, int> mp;  // 定义一个 map，用于存储数字和及其出现次数
    for (int i = 0; i < n; i ++)  // 循环 n 次，处理每个数字
    {
        int t;  // 定义变量 t，表示当前输入的数字
        cin >> t;  // 输入当前数字
        
        string s = to_string(t);  // 将数字 t 转换为字符串 s，方便逐位处理
        int sum = 0;  // 定义变量 sum，用于存储当前数字的各位数字之和
        for (int j = 0; j < s.size(); j ++)  // 遍历字符串 s 的每一位
        {
            sum += s[j] - '0';  // 将字符转换为数字并累加到 sum 中
        }
        
        // 检查当前数字和 sum 是否已经存在于 map 中
        if (mp.find(sum) == mp.end())  // 如果 sum 不存在于 map 中
        {
            mp[sum] = 1;  // 将 sum 插入 map，并设置值为 1
            num ++;  // 不同数字和的个数加 1
        }
    }
    
    // 输出不同数字和的个数
    cout << num << endl;
    
    // 输出所有不同的数字和
    int flag = 1;  // 定义变量 flag，用于控制输出格式（是否输出空格）
    for (auto i : mp)  // 遍历 map 中的每个键值对
    {
        if (flag == 0) cout << " ";  // 如果不是第一个数字和，输出空格
        else flag = 0;  // 将 flag 设置为 0，表示已经输出过第一个数字和
        cout << i.first;  // 输出当前数字和
    }
    cout << endl;  // 输出换行
    return 0;
}