LeetCode-快乐数

题目描述：

编写一个算法来判断一个数是不是“快乐数”。

一个“快乐数”定义为：对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和，然后重复这个过程直到这个数变为 1，也可能是无限循环但始终变不到 1。如果可以变为 1，那么这个数就是快乐数。

示例:

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

 

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/happy-number
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

这道题目的难点在于如何判断是否存在循环。假设输入数字为n，经过第k次计算后得到数字Ak，我们很容易想到，如果Ak不等于1，并且Ak==Ai (i>=0 and i<k)，那么此时进入循环，这个数字一定不是快乐数。根据这个思路，我们很容易想到解法1：

使用一个链表记录从A0到Ak这k+1个数，通过Ak计算出A(k+1)，判断A(k+1)是否在链表中，如果在，表明进入了循环，不是快乐数，否则通过m次计算，Am一定等于1。这个算法的问题在于，如果存在循环，并且k很大时，会占用极大的空间，并且每次计算Ai都要在链表中检查是否存在，很耗时。

#解法1
class Solution(object):
    def isHappy(self, n):
        if n==0:
            return False

        listVals = []
        while True:
            if n==1:
                return True
            try:
                k = listVals.index(n)
                return False
            except:
                listVals.append(n)
                n = self._calcNext(n)
    
    def _calcNext(self, n):
        t = 0
        while n>0:
            x = n % 10
            n = n /10
            t+=x * x
        return t

我们进一步进行抽象，将A0到Am这m个计算结果看成一个链表，其中A(k-1)指向A(k)。我们采用判断链表是否是存在环的解法，快慢指针思路来解决这个问题。如果链表存在环，那么经过有限步数后，快慢指针指向的位置相同。此时，判断指向的值是否等于1，即可断定输入的数字是否为快乐数。解法2代码如下：

#解法2，快慢指针指向同一个位置时，判断值是否等于1
class Solution(object):
    def isHappy(self, n):
        fast = n
        slow = n
        while True:
            fast = self._calcNext(fast)
            fast = self._calcNext(fast)
            slow = self._calcNext(slow)
            if fast==slow:
                return fast==1
        
    def _calcNext(self, n):
        t = 0
        while n>0:
            x = n % 10
            n = n /10
            t += x * x
        return t

解法2的空间复杂度为 O(1)