浙江大学计算机与软件学院 2019 年保研机考 7-1 Happy Numbers (C++)

A happy number is defined by the following process: Starting with any positive integer, replace the number by the sum of the squares of its digits in base-ten, and repeat the process until the number either equals 1 (where it will stay), or it loops endlessly in a cycle that does not include 1. Those numbers for which this process ends in 1 are happy numbers and the number of iterations is called the degree of happiness, while those that do not end in 1 are unhappy numbers (or sad numbers). (Quoted from Wikipedia)

For example, 19 is happy since we obtain 82 after the first iteration, 68 after the second iteration, 100 after the third iteration, and finally 1. Hence the degree of happiness of 19 is 4.

On the other hand, 29 is sad since we obtain 85, 89, 145, 42, 20, 4, 16, 37, 58, and back to 89, then fall into an endless loop. In this case, 89 is the first loop number for 29.

Now your job is to tell if any given number is happy or not.

Input Specification:

Each input file contains one test case. For each case, the first line gives a positive integer $N(\le 100)$. Then $N$ lines follow, each contains a positive integer (no more than $10^4$) to be tested.

Output Specification:

For each given number, output in a line its degree of happiness if it is happy, or the first loop number if it is sad.

Sample Input:

3
19
29
1

Sample Output:

4
89
0

Solution:

// Talk is cheap, show me the code
// Created by Misdirection 2021-09-19 18:53:15
// All rights reserved.

#include <iostream>
#include <vector>
#include <unordered_map>

using namespace std;

int firstLoop, degree;

bool isHappy(int a){
    int cnt = 1;
    if(a == 1){
        degree = 0;
        return true;
    }
    
    unordered_map<int, bool> flag;

    while(true){
        if(flag[a]){
            firstLoop = a;
            return false;
        }

        flag[a] = true;

        int tmp = 0;
        while(a != 0){
            tmp += (a % 10) * (a % 10);
            a /= 10;
        }

        if(tmp == 1){
            degree = cnt;
            return true;
        }
        a = tmp;
        cnt++;
    }
}

int main(){
    int n;
    scanf("%d", &n);

    int num;
    for(int i = 0; i < n; ++i){
        scanf("%d", &num);

        if(isHappy(num)) printf("%d\n", degree);
        else printf("%d\n", firstLoop);
    }

    return 0;
}