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 几种取整的方法