count_digit.py

from collections import Counter

class Solution:
    def count_digits(self, nums):
        for i, num in enumerate(nums):
            nums[i] = self._calculate(num)

        count_freq = Counter(nums)
        max_freq = max(count_freq.values())
        candidates = [num for num, freq in count_freq.items() if freq == max_freq]

        return max(candidates)

    def _calculate(self, num):
        num_str = str(num)

        while len(num_str) > 1:
            total = 0
            for digit in num_str:
                total += int(digit)

            num_str = str(total)

        return int(num_str)

if __name__ == "__main__":
    solution = Solution()
    result = solution.count_digits(["123", "456", "29", "92", "1"])
    print(result)
    print("Test Case Passed!")