LeetCode633 | Sum of Square Numbers (Easy)

题目地址(633. 平方数之和)

https://leetcode-cn.com/problems/sum-of-square-numbers/

题目描述

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

 

示例 1：

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


示例 2：

输入：c = 3
输出：false


示例 3：

输入：c = 4
输出：true


示例 4：

输入：c = 2
输出：true


示例 5：

输入：c = 1
输出：true

 

提示：

0 <= c <= 231 - 1

前置知识

• Java中Math函数的使用

int right = (int) Math.sqrt(c);

思路

• 可以看成是在元素为 0~target 的有序数组中查找两个数，使得这两个数的平方和为 target，如果能找到，则返回 true，表示 target 是两个整数的平方和。
• 本题和 167. Two Sum II - Input array is sorted 类似，只有一个明显区别：一个是和为 target，一个是平方和为 target。本题同样可以使用双指针得到两个数，使其平方和为 target。
• 本题的关键是右指针的初始化，实现剪枝，从而降低时间复杂度。设右指针为 x，左指针固定为 0，为了使 02 + x2 的值尽可能接近 target，我们可以将 x 取为 sqrt(target)。
• 因为最多只需要遍历一次 0~sqrt(target)，所以时间复杂度为 O(sqrt(target))。又因为只使用了两个额外的变量，因此空间复杂度为 O(1)。

关键点

• 先控制范围+双指针
• 注意范围上限可使用Math.sqrt©

代码

• 语言支持：Java

Java Code:

// import math;

class Solution {
    public boolean judgeSquareSum(int c) {
        // if(c<0) {return false;}
        int left = 0;
        int right = (int) Math.sqrt(c);
        while (left <= right) {
            int sum = left*left + right*right;
            if (sum == c) {
                return true;
            } else if (sum > c) {
                right--;
            } else {
                left++;
            }
        }
        return false;
    }
}

复杂度分析

令 c 为数值参数。

• 时间复杂度：$O(sqrt(c))$
• 空间复杂度：$O(1)$

---

大佬之作