leetcode 633. 平方数之和(python)

最新推荐文章于 2021-04-28 22:14:07 发布

洛洛洛洛洛洛洛

最新推荐文章于 2021-04-28 22:14:07 发布

阅读量684

点赞数

分类专栏： leetcode 文章标签： LeetCode

本文链接：https://blog.csdn.net/weixin_44740082/article/details/91392756

版权

该博客主要介绍了LeetCode第633题的解决方案。通过设置两个指针begin和end，分别从0和目标值c的平方根开始，检查它们平方和是否等于c。当平方和大于c时，end指针前移，反之begin指针后移。以此方式遍历，判断是否存在满足条件的整数a和b。

摘要由CSDN通过智能技术生成

题目链接

题目描述：
给定一个非负整数 c ，你要判断是否存在两个整数 a 和 b，使得 a2 + b2 = c。

示例1:

输入: 5
输出: True
解释: 1 * 1 + 2 * 2 = 5

示例2:

输入: 3
输出: False

解题思路：

设置两个索引begin，end分别从0，c的平方根处开始，计算这两个数的平方和，比c大，end减1，比c小，begin加1.

class Solution(object):
    def judgeSquareSum(self, c):
        """
        :type c: int
        :rtype: bool
        """
        begin,end=0,int(math.sqrt(c))
        while begin<=end:
            if begin**2+end**2>c:
                end-=1
            elif begin**2+end**2<c:
                begin+=1
            else:
                return  True
        return False