Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import java.util.Arrays;
- public class ColorSort {
- enum Color {red, green, blue}
- ;
- public static void main(String[] param) {
- Color[] balls = {Color.BLUE, Color.RED, Color.GREEN, Color.BLUE, Color.RED, Color.GREEN, Color.BLUE, Color.RED, Color.GREEN};
- sortBalls(balls);
- System.out.println(Arrays.toString(balls));
- }
- public static void reorder(Color[] balls) {
- int red = 0, green = 0, blue = 0;
- for (Color ball : balls) {
- switch (ball) {
- case RED -> red++;:
- case GREEN -> green++;:
- case BLUE -> blue++;:
- default -> throw new IllegalArgumentException("Incorrect color");
- }
- }
- for (int i = 0; i < red; i++) {
- balls[i] = Color.RED;
- }
- for (int i = red; i < red + green; i++) {
- balls[i] = Color.GREEN;
- }
- for (int i = red + green; i < red + green + blue; i++) {
- balls[i] = Color.BLUE;
- }
- }
- }
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement