Sum_Of_Square_Numbers.js

/*
Sum of Square Numbers
https://leetcode.com/problems/sum-of-square-numbers/description/

Given a non-negative integer c, your task is to decide whether there're two integers a and b such that a2 + b2 = c.

Example 1:
Input: 5
Output: True
Explanation: 1 * 1 + 2 * 2 = 5

Example 2:
Input: 3
Output: False
*/

/**
 * @param {number} c
 * @return {boolean}
 */
var judgeSquareSum = function (c) {
  var iter = 0;
  var set = new Set();
  while (iter ** 2 <= c) {
    var square = iter * iter;
    if (square * 2 === c || set.has(c - square)) return true;

    set.add(square);
    iter++;
  }

  return false;
};

module.exports.judgeSquareSum = judgeSquareSum;