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

public static int[] what(int[] arr, int exp) {
    int n = arr.length;
    int[] output = new int[n];
    int[] count = new int[10];
    for (int i = 0; i < n; i++) {
        int digit = (arr[i] / exp) % 10;
        count[digit]++;
    }
    // ***** סיום שלב 1 *****
    for (int i = 1; i < 10; i++) {
        count[i] += count[i - 1];
    }
    // ***** סיום שלב 2 *****
    for (int i = n - 1; i >= 0; i--) {
        int digit = (arr[i] / exp) % 10;
        output[count[digit] - 1] = arr[i];
        count[digit]--;
    }
    // ***** סיום שלב 3 *****
    return output;
}