LeetCode - 633 - 平方数之和（sum-of-square-numbers）

最新推荐文章于 2023-02-25 22:41:55 发布

JavaScriptLiang

最新推荐文章于 2023-02-25 22:41:55 发布

阅读量196

点赞数

本文链接：https://blog.csdn.net/weixin_41806099/article/details/103342425

版权

一目录

不折腾的前端，和咸鱼有什么区别

目录
一目录
二前言
三解题及测试
四 LeetCode Submit
五解题思路
六进一步思考

二前言

难度：简单
涉及知识：数学
题目地址：https://leetcode-cn.com/problems/sum-of-square-numbers/
题目内容：

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

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

示例2:
输入: 3
输出: False

三解题及测试

小伙伴可以先自己在本地尝试解题，再回来看看 jsliang 的解题思路。

LeetCode 给定函数体：

/**
 * @param {number} c
 * @return {boolean}
 */
var judgeSquareSum = function(c) {
  
};

根据上面的已知函数，尝试破解本题吧~

确定了自己的答案再看下面代码哈~

index.js

/**
 * @name 平方数之和
 * @param {number} c
 * @return {boolean}
 */
const judgeSquareSum = (c) => {
  const sqrtNumber = Math.ceil(Math.sqrt(c));
  const map = new Map();
  for (let i = -sqrtNumber; i < sqrtNumber; i++) {
    const squareNumber = i * i;
    if (map.get(squareNumber) !== undefined) {
      return true;
    }
    map.set(c - squareNumber, i);
  }
  return c === 0 || false;
};

console.log(judgeSquareSum(5)); // true
console.log(judgeSquareSum(3)); // false
console.log(judgeSquareSum(2)); // true
console.log(judgeSquareSum(0)); // true

node index.js 返回：

true
false
true
true

四 LeetCode Submit

Accepted
* 124/124 cases passed (180 ms)
* Your runtime beats 8.46 % of javascript submissions
* Your memory usage beats 6.9 % of javascript submissions (61.6 MB)

五解题思路

首先，拿到题目，不要慌乱，开始尝试：

const judgeSquareSum = (c) => {
  const map = new Map();
  for (let i = 1; i < 5; i++) {
    if (
      map.get(i * i) !== undefined
      || i * i === c
    ) {
      return true;
    }
    if (i * i > c) {
      break;
    }
    map.set(c - i * i, i);
  }
  return false;
};

在这里，我们做的工作：

设置 Map 记录哈希。
遍历 1 —— c。
每个元素我们都设置它期待的值 c - i * i 为 i。例如 c 为 5 时，当遍历到 1 时，期待 4。
判断当前的 i * i 是否出现了预期值，如果有，则返回 true。
判断当前的 i * i 是否大于 c，如果是，说明循环不应持续下去了。

Submit 提交：

Wrong Answer
73/124 cases passed (N/A)

Testcase
2

Answer
false

Expected Answer
true

正常翻车，仔细看题：

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

意思就是这两个整数 a 和 b，还可能是负整数和 0 咯，调整代码：

const judgeSquareSum = (c) => {
  const sqrtNumber = Math.ceil(Math.sqrt(c));
  const map = new Map();
  for (let i = -sqrtNumber; i < sqrtNumber; i++) {
    const squareNumber = i * i;
    if (map.get(squareNumber) !== undefined) {
      return true;
    }
    map.set(c - squareNumber, i);
  }
  return c === 0 || false;
};

然后，Submit 提交：

Accepted
* 124/124 cases passed (180 ms)
* Your runtime beats 8.46 % of javascript submissions
* Your memory usage beats 6.9 % of javascript submissions (61.6 MB)

最后，讲讲解题思路：

众所周知如果不希望两个数的平方和大于某个数，那么它的范围应该是这个数开方后的数字。所以：Math.ceil(Math.sqrt(c))。
使用 map 记录出现的数字期待的值。
遍历 [-sqrtNumber, -sqrtNumber] 的范围，如果出现了哈希表匹对的数字，那么就是成功的。
return 的时候，还需要判断特殊条件 0 的情况。

六进一步思考

得到上面 8.46% 和 6.9% 的答案，证明我这个哈希记录已经很失败了，思考下为什么：

如果这个数特别大，例如 10000，那么我们查找的范围是 [-100, 100]，即 200 次。如果这个数再大点，那遍历的次数就多了。
从小到大遍历，找的顺序是固定的，例如 10000，查找的是：-100, -99, -98, ...0, 1, 2, ...98, 99, 100。观察数字，你应当发现，我们做了很多无用功。

第一次优化

const judgeSquareSum = (c) => {
  if (Number.isInteger(Math.sqrt(c))) {
    return true;
  }
  const sqrtNumber = Math.ceil(Math.sqrt(c));
  for (let i = 0; i < sqrtNumber; i++) {
    if (
      Number.isInteger(Math.sqrt(c - i * i))
      || i * i * 2 === c
    ) {
      return true;
    }
  }
  return false;
};

Submit 提交：

Accepted
* 124/124 cases passed (72 ms)
* Your runtime beats 70.77 % of javascript submissions
* Your memory usage beats 20.69 % of javascript submissions (34.8 MB)

比起优化前的，稍微争气了一点点，继续尝试：

第二次优化

const judgeSquareSum = (c) => {
  for (let i = 0; i * i <= c; i++) {
    if (Number.isInteger(Math.sqrt(c - i * i))) {
      return true;
    }
  }
  return false;
};

Submit 提交：

Accepted
* 124/124 cases passed (60 ms)
* Your runtime beats 96.15 % of javascript submissions
* Your memory usage beats 20.69 % of javascript submissions (34.8 MB)

小脑袋瓜想不出更好的了，如果小伙伴有更好的思路或者想法，欢迎评论吐槽或者直接私聊~

不折腾的前端，和咸鱼有什么区别！

jsliang 会每天更新一道 LeetCode 题解，从而帮助小伙伴们夯实原生 JS 基础，了解与学习算法与数据结构。

浪子神剑 会每天更新面试题，以面试题为驱动来带动大家学习，坚持每天学习与思考，每天进步一点！

扫描上方二维码，关注 jsliang 的公众号（左）和 浪子神剑 的公众号（右），让我们一起折腾！

jsliang 的文档库由梁峻荣采用知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议进行许可。
基于https://github.com/LiangJunrong/document-library上的作品创作。
本许可协议授权之外的使用权限可以从 https://creativecommons.org/licenses/by-nc-sa/2.5/cn/ 处获得。