python 算法題——快樂數的多種解法
題目描述:
編寫一個算法來確定一個數字是否“快樂”。 快樂的數字按照如下方式確定:從一個正整數開始,用其每位數的平方之和取代該數,並重復這個過程,直到最後數字要麼收斂等於1且一直等於1,要麼將無休止地循環下去且最終不會收斂等於1。能夠最終收斂等於1的數就是快樂的數字。
例如:19是一個快樂數字,計算過程如下:
1^2+9^2=82
8^2+2^2=68
6^2+8^2=100
1^2+0^2+0^2=1
要求:當輸入快樂的數字時,輸出True,否則輸出False。
思路:
1. 當輸入的不是快樂數字時,會陷入一個無限循環,因此增加一個計數器 count 用來統計計算次數。設定當 count 達到2000次時,認為該數字不是快樂數字,跳出循環結束計算。
2. 因為不確定輸入的數字會是幾位數,因此不采用除法和取模的方法來獲得數字的每一位數,而是利用 for 循環獲取字符串類型數字的每一位來計算平方和。
代碼:
#快樂的數字 def getSumofSquares(num): numStr=str(num) #將待計算的數字轉換成字符串類型 sum=0 digitls=[int(x) for x in numStr] #從字符串中提取出每一位數字存入一個列表 #註:該步略顯多餘,因為python中字符串可以和列表一樣切片取值或循環,見下方更新部分 #print(digitls) for i in digitls: sum += i**2 return sum def main(): n = input() #輸入一個正整數 sumofSqrs = eval(n) count = 0 while sumofSqrs != 1: sumofSqrs = getSumofSquares(sumofSqrs) count += 1 if count > 2000: #當計算次數超過2000次時,跳出循環結束計算 print("False") break else: print("True") main()
改良版
根據網友在評論區提出的不快樂的數字最終會在 [4,16,37,58,89,145,42,20] 這些數字中無限循環,因此可以加入判斷,當數字變為這些數字裡的任意一個(比如4)時就結束循環,輸出False,從而避免無限循環的產生。
修改後的代碼:
#(新)快樂的數字 def getSumofSquares(num): numStr=str(num) sum=0 for i in numStr: sum += int(i)**2 return sum def main(): n = input() #n為一個正整數 sumofSqrs = eval(n) while sumofSqrs != 1 and sumofSqrs != 4: #或 while sumofSqrs not in [1,4,16,37,58,89,145,42,20] sumofSqrs = getSumofSquares(sumofSqrs) else: if sumofSqrs == 1: print("True") else: print("False") main()
采用遞歸
def happy(n): try: if n==1: print('True') else: new = str(n) sum = 0 for c in new: sum += int(c)**2 return happy(sum) except Exception as e: print('False') # print(e) n = eval(input()) happy(n)
數學方法
d = {} while True: m = 0 while n > 0: m += (n%10)**2 n //= 10 if m in d: return False if m == 1: return True d[m] = m n = m
優化過的
class Solution(object): def isHappy(self, n): """ :type n: int :rtype: bool """ record = [] sq_sum = 0 se_n = n while se_n != 1: sq_sum = 0 while se_n > 0: sq_sum += (se_n % 10) * (se_n % 10) se_n = se_n / 10 if sq_sum in record: return False record.append(sq_sum) se_n = sq_sum return True
以上就是python 算法題——快樂數的多種解法的詳細內容,更多關於python 算法題快樂數的資料請關註WalkonNet其它相關文章!
推薦閱讀:
- C++實現LeetCode(202.快樂數)
- python中內置函數ord()返回字符串的ASCII數值實例詳解
- python不等於運算符的具體使用
- Python編程基礎之函數和模塊
- Python 中的判斷語句,循環語句,函數