LeetCode演算法題202:快樂數解析
阿新 • • 發佈:2018-11-11
編寫一個演算法來判斷一個數是不是“快樂數”。
一個“快樂數”定義為:對於一個正整數,每一次將該數替換為它每個位置上的數字的平方和,然後重複這個過程直到這個數變為 1,也可能是無限迴圈但始終變不到 1。如果可以變為 1,那麼這個數就是快樂數。
示例:
輸入: 19
輸出: true
解釋:
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1
這個題的思路是這樣的,如果是快樂數那麼最後一定是要為1的,如果不是快樂數,那麼一定是迴圈的,也就是說算出來的數是會有重複的。所以就可以用一個容器來進行儲存,這裡使用set,每次計算完新的值之後在set中找是否有這個數,如果有那就一定不是快樂數,如果出現了1,那就是快樂數。
C++原始碼:
class Solution {
public:
bool isHappy(int n) {
set<int> s;
while(n != 1)
{
int temp = 0;
while(n)
{
temp += pow(n%10, 2);
n = n / 10;
}
if (s.count(temp)==0)
{
s. insert(temp);
n = temp;
}
else return false;
}
return true;
}
};
還有一種思路和快樂數的性質有關,那就是不是快樂數的數計算過程中一定會出現4這個數,所以這樣更簡單了,如果出現1就是快樂數,如果出現4就不是快樂數。
python3原始碼:
class Solution:
def isHappy(self, n):
"""
:type n: int
:rtype: bool
"""
while n!=1 and n!=4:
temp = 0
while n:
temp += (n % 10)**2
n = n // 10
n = temp
return n==1