-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathHappyNumber.py
More file actions
executable file
·61 lines (45 loc) · 1.37 KB
/
HappyNumber.py
File metadata and controls
executable file
·61 lines (45 loc) · 1.37 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
"""
Write an algorithm to determine if a number n is happy.
A happy number is a number defined by the following process:
Starting with any positive integer, replace the number by the sum of the squares of its digits.
Repeat the process until the number equals 1 (where it will stay), or it loops endlessly in a cycle which does not include 1.
Those numbers for which this process ends in 1 are happy.
Return true if n is a happy number, and false if not.
Example 1:
Input: n = 19
Output: true
Explanation:
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1
Example 2:
Input: n = 2
Output: false
Constraints:
1 <= n <= 231 - 1
"""
class Solution:
def isHappy(self, n: int) -> bool:
def get_sum_of_squares(num):
total = 0
while num > 0:
digit = num % 10
total += digit * digit
num //= 10
return total
seen_numbers = set()
while n != 1:
n = get_sum_of_squares(n)
if n in seen_numbers:
return False # Detected a cycle
seen_numbers.add(n)
return True
# Create an instance of the Solution class
solution = Solution()
# Test cases
test_cases = [19, 2, 1, 100, 68]
for n in test_cases:
is_happy = solution.isHappy(n)
print(f"Number {n} is happy: {is_happy}")
print(f"Number {n} is happy: {is_happy}")