LeetCode 633.平方數之和（python題解）

題目

給定一個非負整數 c ，你要判斷是否存在兩個整數 a 和 b，使得 a² + b² = c

示例

輸入: 5
輸出: True
解釋: 1 * 1 + 2 * 2 = 5

分析

本題與上一題 LeetCode 167. 兩數之和 II 非常相似，把兩數之和變成了平方數之和，利用上一題答案的思路，同樣設置兩個一左一右的指針，很容易就可以解答出來。這裏要注意題目中b的取值範圍，b的最大值應是c的平方根。

題解

class Solution:
    def judgeSquareSum(self, c: int) -> bool:
        sqrt = int(c ** 0.5)
        a = 0
        b = sqrt
        while a <= b:
            sum = a**2 + b**2
            if sum == c:
                return True
            elif sum < c:
                a += 1
            else:
                b -= 1
        return False

用到的基礎知識

計算平方和平方根

1. 使用內置模塊math

import math

math.pow(4,2) 求4的平方

math.sqrt(16) 求16的開平方

2. 使用表達式法

4 ** 2 計算4的平方

16 ** 0.5 計算16的開平方

3. 使用內置函數

pow(4,2) 計算4的平方

pow(16,0.5) 計算16的平方根

取整

1. 向下取整
向下取整直接用內建的 int() 函數：

>>> a = 3.75
>>> int(a)
3

2. 四捨五入

用 round() 函數：

>>> round(3.25); round(4.85)
3.0
5.0

用 math 模塊中的 ceil() 方法：

3. 向上取整
用 math 模塊中的 modf() 方法，該方法返回一個包含小數部分和整數部分的二元組：

>>> import math
>>> math.ceil(3.25)
4.0
>>> math.ceil(3.75)
4.0
>>> math.ceil(4.85)
5.0

4.分別取小數和整數部分

>>> import math
>>> math.modf(3.25)
(0.25, 3.0)
>>> math.modf(3.75)
(0.75, 3.0)
>>> math.modf(4.2)
(0.20000000000000018, 4.0)

參考

【1】CyC2018 Leetcode 題解 - 雙指針
【2】python計算平方和平方根的方法
【3】Python 幾種取整的方法