LeetCode算法题-Sum of Square Numbers（Java实现）

最新推荐文章于 2022-04-09 16:54:48 发布

程序员小川

最新推荐文章于 2022-04-09 16:54:48 发布

阅读量181

点赞数

CC 4.0 BY-SA版权

分类专栏：【算法】---LeetCode 文章标签： LeetCode 算法题面试题

本文链接：https://blog.youkuaiyun.com/codeMas/article/details/88545126

【算法】---LeetCode 专栏收录该内容

274 篇文章

订阅专栏

本文解析了LeetCode上编号为144的算法题目，介绍了四种不同的解题方法，包括暴力解法、使用HashSet、不使用HashSet的数学解法以及双指针解法，每种方法都提供了详细的Java代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这是悦乐书的第276次更新，第292篇原创

01 看题和准备

今天介绍的是LeetCode算法题中Easy级别的第144题（顺位题号是633）。给定一个非负整数c，判断是否存在两个整数a和b，使得a的平方与b的平方之和等于c。例如：

输入：5
输出：true
说明：1 x 1 + 2 x 2 = 5

输入：3
输出：false

本次解题使用的开发工具是eclipse，jdk使用的版本是1.8，环境是win7 64位系统，使用Java语言编写和测试。

02 第一种解法

暴力解法，直接使用两层for循环，分别从0开始，上限为c的平方根，如果存在两数平方和等于c，就返回true，否则返回false。

此解法可能会超时，不建议使用。

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

03 第二种解法

使用HashSet。如果在c的平方根范围内，存在两数平方和等于c，那么先将单个数的平方值a添加进set中，然后再去判断set中是否存在c减去a的另一个值b，如果存在就返回true。

public boolean judgeSquareSum(int c) {
    HashSet<Integer> set = new HashSet<>();
    int num = (int)Math.sqrt(c);
    for (int i=num; i >= 0; i--) {
        set.add(i*i);
        if (set.contains(c-i*i)) {
            return true;
        }
    }
    return false;
}

04 第三种解法

我们也可以不使用HashSet。依旧是先确定取值范围，上限为c的平方根取整。在0到c的平方根范围内，如果当前一个数的平方根正好等于c，直接返回true，因为另外一个数可能是0；如果不等于，就用c减去当前此数的平方根并赋值给a，再对得到的差开方并赋值给b，如果b的平方等于a，直接返回true，说明存在两数之和等于c。

public boolean judgeSquareSum(int c) {
    int num = (int)Math.sqrt(c);
    for (int i=num; i >= 0; i--) {
        if (i*i == c) {
            return true;
        }
        int a = c - i*i;
        int b = (int)Math.sqrt(a);
        if (b*b == a) {
            return true;
        }
    }
    return false;
}

05 第四种解法

使用双指针。首指针a从0开始，尾指针b从c的平方根开始，如果a的平方加上b的平方的值大于c，那么尾指针b就减1；如果小于c，那么首指针a就加1；如果等于c，直接返回true。

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