נתונה הפונקציה what המקבלת רשימה של מספרים שלמים חיוביים arr ופרמטר נוסף exp. הפרמטר exp יכול לקבל ערכים שהם חזקות של 10 (כגון 1, 10, 100, 1000...).

def what(arr, exp):
    n = len(arr)
    output = [0] * n
    count = [0] * 10
    for i in range(n):
        digit = (arr[i] // exp) % 10
        count[digit] += 1
    # **** סיום שלב 1 ****
    for i in range(1, 10):
        count[i] += count[i - 1]
    # **** סיום שלב 2 ****
    for i in range(n - 1, -1, -1):
        digit = (arr[i] // exp) % 10
        output[count[digit] - 1] = arr[i]
        count[digit] -= 1
    # **** סיום שלב 3 ****
    return output