Demo20快乐数

/**
 * 编写一个算法来判断一个数 n 是不是快乐数。
 * 「快乐数」 定义为：
 * 对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
 * 然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
 * 如果这个过程 结果为 1，那么这个数就是快乐数。
 * 如果 n 是 快乐数 就返回 true ；不是，则返回 false 。
 */

public class Demo20 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        System.out.println("请输入一个正整数");
        int n = sc.nextInt();
        if(n==1){
            System.out.println(n+"是快乐数");
        }
        Set<Integer> seen = new HashSet<>();
        while (!seen.contains(n)) {
            seen.add(n);
            n =getNext(n);
            if(n==1){
                System.out.println(n+"是快乐数");
                return;
            }
        }
        System.out.println(n+"不是快乐数");
    }


    public static int getNext(int n) {
        int totalSum = 0;

        while (n > 0) {
            totalSum += Math.pow(n % 10, 2);
            n /= 10;
        }

        return totalSum;
    }
}