From aaf6f04d5f55975bf78cc644d9f944081472fe71 Mon Sep 17 00:00:00 2001
From: SKTT1Ryze <ccc@neos.moe>
Date: Mon, 6 May 2024 16:32:54 +0800
Subject: [PATCH 1/2] add perfect squares

---
 leetcode-cc/PerfectSquares.hpp | 29 +++++++++++++++++++++++++++++
 runtime-cc/src/registration.cc |  6 ++++++
 2 files changed, 35 insertions(+)
 create mode 100644 leetcode-cc/PerfectSquares.hpp

diff --git a/leetcode-cc/PerfectSquares.hpp b/leetcode-cc/PerfectSquares.hpp
new file mode 100644
index 0000000..18f6ba8
--- /dev/null
+++ b/leetcode-cc/PerfectSquares.hpp
@@ -0,0 +1,29 @@
+#include "TestHelper.h"
+#include "problem.h"
+#include "solution.h"
+
+using namespace std;
+
+IMPLEMENT_PROBLEM_CLASS(PPerfectSquares, 279, DIFFI_MEDIUM, TOPIC_ALGORITHMS,
+                        "Perfect Squares",
+                        "Given an integer n, return the least number of "
+                        "perfect square numbers that sum to n.",
+                        {"Math"});
+
+class SPerfectSquares : public ISolution {
+ public:
+  size_t problemId() const override { return 279; }
+  string name() const override {
+    return ("Solution for " + string("Perfect Squares"));
+  }
+  string location() const override { return __FILE_NAME__; }
+  int test() const override {
+    return testHelper<int, int>({12, 13}, {3, 2}, [this](auto input) {
+      return this->numSquares(input);
+    });
+  };
+  int benchmark() const override { return 0; }
+
+ private:
+  int numSquares(int n) const {}
+};
diff --git a/runtime-cc/src/registration.cc b/runtime-cc/src/registration.cc
index 605e13a..062c930 100644
--- a/runtime-cc/src/registration.cc
+++ b/runtime-cc/src/registration.cc
@@ -94,6 +94,7 @@
 #include "../leetcode-cc/PascalsTriangleII.hpp"
 #include "../leetcode-cc/PathSum.hpp"
 #include "../leetcode-cc/PathSumII.hpp"
+#include "../leetcode-cc/PerfectSquares.hpp"
 #include "../leetcode-cc/PopulateNextRightPointers.hpp"
 #include "../leetcode-cc/ProductOfArrayExceptSelf.hpp"
 #include "../leetcode-cc/PseudoPalindromicPathsInBTree.hpp"
@@ -982,5 +983,10 @@ const int registerAll(std::shared_ptr<Container> handle) {
     return std::make_shared<SProductOfArrayExceptSelf>();
   });
 
+  handle->registerProblem(
+      []() -> ArcProblem { return std::make_shared<PPerfectSquares>(); });
+  handle->registerSolution(
+      []() -> ArcSolution { return std::make_shared<SPerfectSquares>(); });
+
   return 0;
 }

From 6116d6d46affa1c64a0a175bcc8ef700cd2f1323 Mon Sep 17 00:00:00 2001
From: SKTT1Ryze <ccc@neos.moe>
Date: Mon, 6 May 2024 16:53:01 +0800
Subject: [PATCH 2/2] impl perfect squares

---
 leetcode-cc/PerfectSquares.hpp | 26 +++++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/leetcode-cc/PerfectSquares.hpp b/leetcode-cc/PerfectSquares.hpp
index 18f6ba8..46a5af6 100644
--- a/leetcode-cc/PerfectSquares.hpp
+++ b/leetcode-cc/PerfectSquares.hpp
@@ -1,3 +1,6 @@
+#include <queue>
+#include <unordered_set>
+
 #include "TestHelper.h"
 #include "problem.h"
 #include "solution.h"
@@ -25,5 +28,26 @@ class SPerfectSquares : public ISolution {
   int benchmark() const override { return 0; }
 
  private:
-  int numSquares(int n) const {}
+  int numSquares(int n) const {
+    queue<pair<int, int>> q = {};
+    unordered_set<int> memo = {};
+
+    q.push({0, 0});
+
+    while (!q.empty()) {
+      auto [val, step] = q.front();
+      q.pop();
+
+      if (val == n) return step;
+
+      memo.insert({val, step});
+
+      for (int i = 1; i * i + val <= n; i++) {
+        auto nextVal = val + i * i;
+        if (!memo.contains(nextVal)) q.push({nextVal, step + 1});
+      }
+    }
+
+    return -1;
+  }
 };