Problem_8_Dutch_National_Flag_Problem.java

package Two_Pointers;

// Problem Statement: Dutch National Flag Problem (medium)
// LeetCode Question: 75. Sort Colors

public class Problem_8_Dutch_National_Flag_Problem {
    public int[] sort(int[] arr){
        int low = 0, high = arr.length - 1;
        for (int i = 0; i <= high;) {
            if (arr[i] == 0) {
                swap(arr, i, low);
                i++;
                low++;
            } else if (arr[i] == 1) {
                i++;
            } else {
                swap(arr, i, high);
                high--;
            }
        }
        return arr;
    }

    private static void swap(int[] arr, int i, int j){
        int temp = arr[i];
        arr[i] = arr[j];
        arr[j] = temp;
    }
}