编写一个算法来判断一个数 n 是不是快乐数。
「快乐数」定义为:对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和,然后重复这个过程直到这个数变为 1,也可能是 无限循环 但始终变不到 1。如果 可以变为 1,那么这个数就是快乐数。
如果 n 是快乐数就返回 True ;不是,则返回 False 。
示例:
输入:19
输出:true
解释:
1^2 + 9^2 = 82
8^2 + 2^2 = 68
6^2 + 8^2 = 100
1^2 + 0^2 + 0^2 = 1
题目中说了会 无限循环,那么也就是说求和的过程中,sum会重复出现,这对解题很重要!
当我们遇到了要快速判断一个元素是否出现集合里的时候,就要考虑哈希法了。
所以这道题目使用哈希法,来判断这个sum是否重复出现,如果重复了就是return false, 否则一直找到sum为1为止。
解法一:
class Solution { public boolean isHappy(int n) { Set<Integer> set=new HashSet<>(); while(n!=1){ //用set来判断是否有循环 if(set.contains(n)){ return false; } set.add(n); //数字转成字符串 String str=n+""; int sum=0; for(char ch:str.toCharArray()){ //字符转数字 int num=ch-'0'; sum+=Math.pow(num,2); } n=sum; } return true; } }
下面这种解法时间上更好
class Solution { public boolean isHappy(int n) { Set<Integer> set=new HashSet<>(); while(n!=1 && !set.contains(n)){ set.add(n); n=getNextNumber(n); } return n==1; } private int getNextNumber(int n){ int sum=0; while(n>0){ //拿到个位数 int temp=n%10; sum+=temp*temp; //拿到十位数 n=n/10; } return sum; } }