Python 练习 双指针法: LeetCode 633 平方数之和

Python 练习 双指针法: LeetCode 633 平方数之和

给定一个非负整数 c ，你要判断是否存在两个整数 a 和 b，使得 a^2 + b^2 = c 。

示例 1：
输入：c = 5
输出：true
解释：1 * 1 + 2 * 2 = 5

示例 2：
输入：c = 3
输出：false

提示：
0 <= c <= 2^31 - 1

---

题解一

两层遍历
采用暴力求解思路的两层遍历，时间复杂度过高

import math
class Solution:
    def judgeSquareSum(self, c: int) -> bool:
        c_sqrt = int(math.sqrt(c))
        for i in range(c_sqrt+1):
            for j in range(i+1):
                if i**2 + j**2 == c:
                    return True
        return False

---

题解二

双指针法
暴力求解的两层遍历过程中，很多种组合情况重复出现，应当避免答案空间的重复搜索。采用双指针法，将两个指针初始化为 0，若指针平方和偏小，增加指针 1 的值，反之，回退指针 1 并让指针 2 前进。如果出现后发的指针 2 值大于指针 1，则说明可能的答案空间已全部搜索，输入参数 c 无法表示为平方和

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

题解三

双指针法 优化 -> 二分法
二分法在这一题中其实还是双指针法，只是将一个指针的起点设置为 sqrt(c)。这样的设置比题解二更合理，可以有效缩小答案空间的搜索范围

class Solution:
    def judgeSquareSum(self, c: int) -> bool:
        left = 0
        right = int(c**0.5)
        while left <= right:
            cur = left * left + right * right
            if cur == c: 
                return True
            if cur < c:
                left += 1
            else:
                right -= 1
        return False

题解四

利用内置 sqrt() 或 ** 0.5
最后一种方法回归了暴力求解的思路，但是与两层循环不同，这里更充分地利用了 sqrt() 方法。由于内置方法时间复杂度优化程度高，因此这种方法实现了最好的性能

import math
class Solution(object):
    def judgeSquareSum(self, c):
        """
        :type c: int
        :rtype: bool
        """
        if c == 0: return True
        for a in range(1, int(math.sqrt(c) + 1)):
            b = c - a * a
            if int(math.sqrt(b)) ** 2 == b:
                return True
        return False

---

小结：

这一题可以通过暴力求解、双指针法、二分法等算法思路完成求解。其中，双指针法可以在较小的搜索空间内完成求解，二分法在其基础上进一步缩小了搜索空间；最后，如果能充分利用内置函数的时间复杂度优势，同样可以实现性能很好的算法。这一题中，时间复杂度为题解三优于题解四优于题解二