1. 程式人生 > >leet202. 快樂數

leet202. 快樂數

題目:

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

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

案例: 19 是一個快樂數。

  • 12 + 92 = 82
  • 82 + 22 = 68
  • 62 + 82 = 100
  • 12 + 02 + 02 = 1

分析:

  1. 分離n的每一位,計算平方和,若為1,則返回真,若不為1則將平方和賦值給n,繼續迴圈;
  2. 使用集合記錄過程中的出現的平方和,若平方和第二次出現,說明陷入死迴圈,返回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)

思考: