05-Sort an array of 0's, 1's and 2's

#include <bits/stdc++.h> 
void sort012(int *arr, int n)
{
   //   Write your code here
   //Brute force
   //sort(arr,arr+n);

   //Better approach
   // int zero=0,one=0,two=0;

   // for(int i=0;i<n;i++){
   //    if(arr[i]==0){
   //       zero++;
   //    }
   //    else if(arr[i]==1){
   //       one++;
   //    }
   //    else{
   //       two++;
   //    }
   // }

   // for(int i=0;i<zero;i++){
   //    arr[i]=0;
   // }

   // for(int i=zero;i<zero+one;i++){
   //    arr[i]=1;
   // }

   // for(int i=zero+one;i<zero+one+two;i++){
   //    arr[i]=2;
   // }

   //optimal approach
   int low = 0, mid = 0, high = n - 1; // 3 pointers

    while (mid <= high) {
        if (arr[mid] == 0) {
            swap(arr[low], arr[mid]);
            low++;
            mid++;
        }
        else if (arr[mid] == 1) {
            mid++;
        }
        else {
            swap(arr[mid], arr[high]);
            high--;
        }
    }
}