快樂數 【LeetCode 集合set的使用】
阿新 • • 發佈:2018-12-15
編寫一個演算法來判斷一個數是不是“快樂數”。
一個“快樂數”定義為:對於一個正整數,每一次將該數替換為它每個位置上的數字的平方和,然後重複這個過程直到這個數變為 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; } } } }