202. 快乐数

题目描述

编写一个算法来判断一个数 n 是不是快乐数。

「快乐数」 定义为：

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

解题思路

#![allow(unused)]
fn main() {
struct Solution {}
impl Solution {
    pub fn is_happy(mut n: i32) -> bool {
        use std::collections::HashSet;
        let mut set = HashSet::new();
        set.insert(n);
        fn f(mut n: i32) -> i32 {
            let mut res = 0;
            while n != 0 {
                let x = n % 10;
                res += x * x;
                n /= 10;
            }
            res
        }
        loop {
            n = f(n);
            if n == 1 { return true }
            if set.contains(&n) { return false }
            set.insert(n);
        }
    }
}
}

学习感想

一下子不知道怎么做，但是把false的例子弄明白就知道了

#![allow(unused)]
fn main() {
struct Solution {}

impl Solution {
    pub fn is_happy(mut n: i32) -> bool {
        use std::collections::HashSet;
        let mut set: HashSet<i32> = HashSet::from([n]);
        loop {
            let mut new = 0i32;
            while n != 0i32 {
                new += (n % 10i32).pow(2u32);
                n /= 10i32;
            }
            if new == 1i32 { break true }
            if set.contains(&new) { break false }
            set.insert(new);
            n = new;
        }
    }
}
}