小猴编程周赛C++ | 2022年秋第7周T3优雅的数

学习C++从娃娃抓起！记录下在学而思小猴编程学习过程中的题目，记录每一个瞬间。侵权即删，谢谢支持！

附上汇总贴：小猴编程C++ | 汇总-CSDN博客

---

【题目描述】
小猴对数字非常敏感，机缘巧合之下，他发现了一种特殊的数字，小猴称其为优雅的数。优雅的数定义为：如果对于一个正整数n，每次将该数替换为它的每个位置上的数字的平方的和，重复这个过程，如果最终这个整数可以变成1，则称这个整数为优雅的数。
例如，数字19就是一个优雅的数，因为：19 -> 82 -> 68 -> 100 -> 1。
现在给定你一个正整数n，请你判断n是否为优雅的数，如果n是优雅的数，输出Yes，否则输出No。
【输入】
第一行，包含一个整数T，表示输入包含T组测试数据；
每组测试数据输入一行，一个整数n。
【输出】
每组测试数据输出一行，如果该数是优雅的数，输出Yes，否则输出No。
【输入样例】

2
7
116

【输出样例】

Yes
No

【代码详解】

#include <bits/stdc++.h>
using namespace std;
bool f[2000];  // 18*81=1458，所以开2000的数组可以满足
long long s_sum(long long x)
{
    long long sum = 0;
    while (x) {  // 数位拆分并计算平方和
        int g = x%10;
        sum += g*g;
        x /=10;
    }
    return sum;
}
int main()
{
    int T;
    cin >> T;
    for (int i=1; i<=T; i++) {
        long long n;
        cin >> n;
        memset(f, 0, sizeof(f));  // 每次数据计算，都要将f数组初始化
        bool flag = true;
        n = s_sum(n);  // 先将n更新为数字的平方和
        while (n!=1) {  // 不等于1时进行重复计算
            f[n] = 1;  // 桶排序，n的位置置为1
            n = s_sum(n);  // 再次进行计算
            if (f[n]==1) {  // 如果判断n这个数字曾经出现过
                flag = false;  // flag置为flase，并退出循环
                break;
            }
        }
        if (flag) cout << "Yes" << endl;  // 如果没有陷入死循环，就肯定有结果
        else cout << "No" << endl;
    }
    return 0;
}

【运行结果】

2
7 
Yes
116
No