Implementation for Trimorphic Number

trimorphic_number.py

@@ -0,0 +1,26 @@
+"""
+A Trimorphic Number is a number whose cube ends in the same digits as the number itself.
+For example, 4^3 = 64, which ends in 4.
+
+Reference: https://en.wikipedia.org/wiki/Automorphic_number#Trimorphic_numbers
+"""
+
+
+def is_trimorphic(number: int) -> bool:
+    """
+    Checks if a number is a Trimorphic number.
+    """
+    if number < 0:
+        raise ValueError("Input must be a non-negative integer")
+
+    if number == 0:
+        return True
+
+    cube = number**3
+    return str(cube).endswith(str(number))
+
+
+if __name__ == "__main__":
+    import doctest
+
+    doctest.testmod()