leet202. 快樂數
阿新 • • 發佈:2018-11-10
題目:
寫一個演算法來判斷一個數是不是“快樂數”。
一個數是不是快樂是這麼定義的:對於一個正整數,每一次將該數替換為它每個位置上的數字的平方和,然後重複這個過程直到這個數變為 1,或是無限迴圈但始終變不到 1。如果可以變為 1,那麼這個數就是快樂數。
案例: 19 是一個快樂數。
- 12 + 92 = 82
- 82 + 22 = 68
- 62 + 82 = 100
- 12 + 02 + 02 = 1
分析:
- 分離n的每一位,計算平方和,若為1,則返回真,若不為1則將平方和賦值給n,繼續迴圈;
- 使用集合記錄過程中的出現的平方和,若平方和第二次出現,說明陷入死迴圈,返回False
程式碼:
class Solution(object): def isHappy(self, n): """ :type n: int :rtype: bool """ setTemp = [] while True: sums = 0 while n // 10: sums += (n % 10) ** 2 n //= 10 sums += (n % 10) ** 2 if sums == 1: return True n = sums # print sums # print setTemp if sums in setTemp: return False setTemp.append(sums)
思考: