快乐数。编写一个算法来判断一个数 n 是不是快乐数。「快乐数」定义为:对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和。然后重复这个过程直到这个数变为 1,也可能是 无限循环 但始终变不到 1。如果 可以变为 1,那么这个数就是快乐数。如果 n 是快乐数就返回 true ;不是,则返回 false 。力扣202。
方法1:自然智慧。用集合。
方法2:最终会遇到1或者4。1是快乐数,4不是快乐数。
代码用golang编写。代码如下:
package main
import "fmt"
func main() {
n := 5
if true {
ret := isHappy1(n)
fmt.Println(ret)
}
if true {
ret := isHappy2(n)
fmt.Println(ret)
}
}
func isHappy1(n int) bool {
set := make(map[int]struct{})
for n != 1 {
sum := 0
for n != 0 {
r := n % 10
sum += r * r
n /= 10
}
n = sum
if _, ok := set[n]; ok {
break
}
set[n] = struct{}{}
}
return n == 1
}
实验代码
//public static TreeSet<Integer> sum(int n) {
//TreeSet<Integer> set = new TreeSet<>();
//while (!set.contains(n)) {
//set.add(n);
//int sum = 0;
//while (n != 0) {
//sum += (n % 10) * (n % 10);
//n /= 10;
//}
//n = sum;
//}
//return set;
//}
func isHappy2(n int) bool {
for n != 1 && n != 4 {
sum := 0
for n != 0 {
sum += (n % 10) * (n % 10)
n /= 10
}
n = sum
}
return n == 1
}
执行结果如下: