diff --git a/problems/0202.快乐数.md b/problems/0202.快乐数.md
index 1c630b6a..2e784e6a 100644
--- a/problems/0202.快乐数.md
+++ b/problems/0202.快乐数.md
@@ -111,25 +111,29 @@ Python：
 ```python
 class Solution:
     def isHappy(self, n: int) -> bool:
-        set_ = set()
-        while 1:
-            sum_ = self.getSum(n)
-            if sum_ == 1:
+        def calculate_happy(num):
+            sum_ = 0
+            
+            # 从个位开始依次取，平方求和
+            while num:
+                sum_ += (num % 10) ** 2
+                num = num // 10
+            return sum_
+
+        # 记录中间结果
+        record = set()
+
+        while True:
+            n = calculate_happy(n)
+            if n == 1:
                 return True
-            #如果这个sum曾经出现过，说明已经陷入了无限循环了，立刻return false
-            if sum_ in set_:
+            
+            # 如果中间结果重复出现，说明陷入死循环了，该数不是快乐数
+            if n in record:
                 return False
             else:
-                set_.add(sum_)
-            n = sum_
-            
-    #取数值各个位上的单数之和
-    def getSum(self, n):
-        sum_ = 0
-        while n > 0:
-            sum_ += (n%10) * (n%10)
-            n //= 10
-        return sum_
+                record.add(n)
+
 ```
 
 Go：