题目
给定一个非负整数 c ,你要判断是否存在两个整数 a 和 b,使得 a2 + b2 = 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 几种取整的方法