Update count-the-number-of-ideal-arrays.cpp

kamyu104 · web-flow · commit 52333ef0b6c1 · 2023-10-08T18:58:06.000+08:00
diff --git a/C++/count-the-number-of-ideal-arrays.cpp b/C++/count-the-number-of-ideal-arrays.cpp
@@ -9,7 +9,7 @@ class Solution {
         const auto& prime_factors = [&](int x) {
             unordered_map<int, int> factors;
             for (const auto& p : primes) {
-                if (x < p) {
+                if (p * p > x) {
                     break;
                 }
                 for (; x % p == 0; x /= p) {

Original file line number	Diff line number	Diff line change
`@@ -9,7 +9,7 @@ class Solution {`
`9`	`9`	`const auto& prime_factors = [&](int x) {`
`10`	`10`	`unordered_map<int, int> factors;`
`11`	`11`	`for (const auto& p : primes) {`
`12`		`- if (x < p) {`
	`12`	`+ if (p * p > x) {`
`13`	`13`	`break;`
`14`	`14`	`}`
`15`	`15`	`for (; x % p == 0; x /= p) {`