題目鏈接:https://leetcode.com/problems/happy-number/ 題目理解:實現isHappy函數,判斷一個正整數是否為happy數 happy數:計算要判斷的數的每一位的平方和,平方和為1則為happy數,不為1則將原數替換為此平方和,繼續上述步驟,直到等於1(為h ...
題目鏈接:https://leetcode.com/problems/happy-number/
題目理解:實現isHappy函數,判斷一個正整數是否為happy數
happy數:計算要判斷的數的每一位的平方和,平方和為1則為happy數,不為1則將原數替換為此平方和,繼續上述步驟,直到等於1(為happy數),或者進入不包含1的無限迴圈(非happy數)。
測試用例:19為happy數:
解題思路:
定義set集合存每次計算得到的需判斷數n,迴圈計算新的原數n,終止條件為n等於1或者set中出現重覆的數字。迴圈體:將n插入到set中,迴圈取n的每一位計算其平方和,並將n等於這個平方和。最後迴圈結束,如果n等於1,則原數為happy數,否則不是。
代碼:
1 class Solution { 2 public: 3 bool isHappy(int n) { 4 set<int> result; 5 int re = 0; 6 while(n!=1&&!result.count(n)){ 7 result.insert(n); 8 re = 0; 9 while(n){ 10 int a = n%10; 11 re+=a*a; 12 n = n/10; 13 } 14 n = re; 15 } 16 if(n==1) return true; 17 else return false; 18 } 19 };View Code
卡住的點:
- 判斷是否出現不為1的死迴圈,需要記錄每次的原數,迴圈終止條件為原數為1,或者原數在記錄的集合中已存在,即出現重覆。
- 現寫的代碼結構中,將原數n插入到集合中,這步應在while迴圈剛進入時插入,不能在迴圈體最後插入。