Järjestys

import java.util.Arrays;
import java.util.Scanner;
import java.util.function.IntBinaryOperator;

public class Jarjestys implements IntBinaryOperator {

	public static void main(String[] args) {
		Scanner reader = new Scanner(System.in);
		int loop = reader.nextInt();
		int[] sortedNumbers = new int[loop];
		int i = 0;
		for (i=0;i<loop;i++) {
			sortedNumbers[i] = reader.nextInt();
		}
		int[] numbers = sortedNumbers.clone();
		Arrays.parallelSort(sortedNumbers);
		//-----------------------------------------
		int[] amounts = new int[1000000];
		
		for (int key : numbers) {
			amounts[key]++;
		}
		Arrays.parallelPrefix(amounts, new Jarjestys());
		
		//-----------------------------------------
		//Replace numbers with indexes
		for (i=0;i<numbers.length;i++) {
			numbers[i] = --amounts[numbers[i]];
		}
		int[] sortOrder = sortArray(numbers);
		int maara = 0;
		for (int num : sortOrder)
			if (num > 0)
				maara++;
		System.out.println(maara);
		String str = "";
		for (int num : sortOrder)
			if (num > 0)
				str += ((num + 1) + " ");
		str = str.trim();
		System.out.println(str);
		reader.close();
	}
	
	private static int[] sortArray(int[] array) { //TODO: case 1 1 2 0
		int[] sortedArray = array.clone();
		Arrays.parallelSort(sortedArray);
		int index, i = 0;
		int[] sortOrder = new int[5*array.length];
		while ((sortOrder[i++] = index = checkIfSorted(array, sortedArray)) != -1) {
			sortOrder[i++] = index = swap(array, index);
			while (index != 0) {
				sortOrder[i++] = index = swap(array, index);
			}
		}
		return sortOrder;
	}
	
	private static int checkIfSorted(int[] array, int[] sortedArray) { //FIXME: add support for same indexes
		int i;
		for (i=0;i<array.length-1;i++) {
			if (array[i] != sortedArray[i])
				return i;
		}
		return -1;
	}
	
	private static int swap(int[] array, int index) {
		int newIndex = array[index];
		array[index] = array[0];
		array[0] = newIndex;
		return newIndex;
	}

	@Override
	public int applyAsInt(int left, int right) {
		return left + right;
	}
}

Test 1

Group: 1

Verdict: WRONG ANSWER

Test 2

Group: 2

Verdict: WRONG ANSWER

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED