Project Euler 62 Solution

DIRECTORY.md

@@ -650,6 +650,8 @@
     * [Sol1](https://github.com/TheAlgorithms/Python/blob/master/project_euler/problem_551/sol1.py)
   * Problem 56
     * [Sol1](https://github.com/TheAlgorithms/Python/blob/master/project_euler/problem_56/sol1.py)
+  * Problem 62
+    * [Sol1](https://github.com/TheAlgorithms/Python/blob/master/project_euler/problem_62/sol1.py)
   * Problem 63
     * [Sol1](https://github.com/TheAlgorithms/Python/blob/master/project_euler/problem_63/sol1.py)
   * Problem 67

project_euler/problem_62/sol1.py

@@ -0,0 +1,64 @@
+"""
+Project Euler 62
+https://projecteuler.net/problem=62
+
+The cube, 41063625 (345^3), can be permuted to produce two other cubes:
+56623104 (384^3) and 66430125 (405^3). In fact, 41063625 is the smallest cube
+which has exactly three permutations of its digits which are also cube.
+
+Find the smallest cube for which exactly five permutations of its digits are
+cube.
+"""
+
+from collections import defaultdict
+
+
+def solution(max_base: int = 5) -> int:
+    """
+    Iterate through every possible cube and sort the cube's digits in
+    ascending order. Sorting maintains an ordering of the digits that allows
+    you to compare permutations. Store each sorted sequence of digits in a
+    dictionary, whose key is the sequence of digits and value is a list of
+    numbers that are the base of the cube.
+
+    Once you find 5 numbers that produce the same sequence of digits, return
+    the smallest one, which is at index 0 since we insert each base number in
+    ascending order.
+
+    >>> solution(2)
+    125
+    >>> solution(3)
+    41063625
+    >>> solution(5)
+    127035954683
+    """
+    freqs = defaultdict(list)
+    num = 0
+
+    while True:
+        digits = get_digits(num)
+        freqs[digits].append(num)
+
+        if len(freqs[digits]) == max_base:
+            base = freqs[digits][0] ** 3
+            return base
+
+        num += 1
+
+
+def get_digits(num: int) -> str:
+    """
+    Computes the sorted sequence of digits of the cube of num.
+
+    >>> get_digits(3)
+    '27'
+    >>> get_digits(99)
+    '027999'
+    >>> get_digits(123)
+    '0166788'
+    """
+    return "".join(sorted(list(str(num ** 3))))
+
+
+if __name__ == "__main__":
+    print(f"{solution() = }")