1. 程式人生 > >快樂數 【LeetCode 集合set的使用】

快樂數 【LeetCode 集合set的使用】

編寫一個演算法來判斷一個數是不是“快樂數”。

一個“快樂數”定義為:對於一個正整數,每一次將該數替換為它每個位置上的數字的平方和,然後重複這個過程直到這個數變為 1,也可能是無限迴圈但始終變不到 1。如果可以變為 1,那麼這個數就是快樂數。

示例: 

輸入: 19
輸出: true
解釋: 
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1

思路:使用set儲存每次迴圈後計算的值,然後判斷這個值是否已經在集合中,如果在集合中,說明進入了無限迴圈,就返回false

import java.util.HashSet;
import java.util.Set;
public class Main{
    public static void main(String[] args) {
        System.out.println(isHappy(12345));
    }

    public static boolean isHappy(int n) {
        Set<Integer>set=new HashSet<>();
        set.add(n);
        int sum;
        while(true){
            if(n==1){  
                return true;
            }

            sum=0;
            while(n>=1){
                int c=n%10;
                n=n/10;
                sum+=c*c;
            }
            n=sum;
            boolean b=set.add(n);  //如果加入set時,集合中沒有該值,就會返回true,
            if(b==false){          //如果有了該值,就會返回false
                return false;
            }
        }
    }
}