Arrays #4 : Improved Dutch Flag solution

improvedDutchFlag.go

func improvedDutchFlag(arr []int, pivotIndex int)[]int{
  pivot := arr[pivotIndex]
  smaller := 0;

  for i:=0; i < len(arr) ; i++ {
    if arr[i] < pivot {
      arr[i], arr[smaller] = arr[smaller] , arr[i]
      smaller++
    }
  }

  larger := len(arr) - 1 
  for i:= len(arr) - 1 ; i >= 0 ; i-- {
    if arr[i] > pivot {
      arr[i], arr[larger] = arr[larger] , arr[i]
      larger--
    }
  }


  return arr
}