question4.java

public class question4 {
    public static void main(String[] args) {
        int[] a = {2, 0, 1, 2, 0, 1};
        System.out.println("Original Array:");
        for (int i : a) {
            System.out.print(i + " ");
        }

        sortColors(a);
        System.out.println("\nSorted Array:");
        for(int i : a) {
            System.out.print(i + " ");
        }
        }

        static void sortColors(int[] a) {
            int low = 0, mid = 0, high = a.length - 1;
            while(mid <= high){
                switch(a[mid]){
                    case 0:
                    int temp0 = a[low];
                    a[low] = a[mid];
                    a[mid] = temp0;
                    low++;
                    mid++;
                    break;
                    case 1:
                    mid++;
                    break;
                    case 2:
                    int temp2 = a[high];
                    a[high] = a[mid];
                    a[mid] = temp2;
                    high--;
                    break;
                }
            }
        }

    }