LeetCode双指针：平方数之和

最新推荐文章于 2024-11-05 21:58:11 发布

雪落山庄

最新推荐文章于 2024-11-05 21:58:11 发布

阅读量58

点赞数

分类专栏： leetcode 文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/qq_45103836/article/details/134576420

版权

leetcode 专栏收录该内容

72 篇文章 0 订阅

订阅专栏

文章讲述了如何使用枚举法和双指针法解决LeetCode中的一个问题，即判断是否存在两个整数a和b，使得a²+b²=c。双指针法通过控制两个指针a和b的范围，有效减少了搜索空间，提高了效率。

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

LeetCode双指针：平方数之和

题目描述

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

示例 1：

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

示例 2：

输入：c = 3
输出：false

解题思路1：枚举法

首先对c进行开方，作为遍历的一个上限，然后从0到根号 $\sqrt{c}$ ,用 $\sqrt{c-a*a}$ 再开根号，判断是否满足a*a+b*a==c

代码1

class Solution {
    public boolean judgeSquareSum(int c) {
        int max = (int)Math.sqrt(c);
        for (int a = 0; a <= max; a++) {
            int b = (int)Math.sqrt(c - a * a);
            if (a * a + b * b == c) return true;
        }
        return false;
    }
}

解题思路2：双指针法

由于 a 和 b 的范围均为 [0, $\sqrt{c}$ ]，因此我们可以使用「双指针」在 [0, $\sqrt{c}$ ] 范围进行扫描：

小于 c ：左边的元素都小于当前元素，只能下移，相当于low++。此时排除的是黄色格子以及左边同行的元素，都小于 c ，所以不会错过正确答案。
大于 c ：下面的元素都大于当前元素，只能左移，相当于high--。此时排除的是黄色格子以及下方同列的元素，都大于 c ，所以不会错过正确答案。

代码2

class Solution {
    public boolean judgeSquareSum(int c) {
        long a = 0, b = (long) Math.sqrt(c);
        while (a <= b) {
            long cur = a * a + b * b;
            if (cur == c) {
                return true;
            } else if (cur > c) {
                b--;
            } else {
                a++;
            }
        }
        return false;
    }
}