Added 72. Sort Colors

Author: LuaanNguyen

medium/75_sort_colors/75_sort_colors.go

@@ -0,0 +1,28 @@
+package sort_colors
+
+func SortColors(nums []int) []int  {
+    low := 0 
+    mid := 0 
+    high := len(nums) - 1
+
+    for mid <= high {
+        if nums[mid] == 0 {
+            temp := nums[mid]
+            nums[mid] = nums[low]
+            nums[low] = temp 
+
+            mid++
+            low++
+        } else if nums[mid] == 1 {
+            mid++
+        } else if nums[mid] == 2 {
+            temp := nums[mid]
+            nums[mid] = nums[high]
+            nums[high] = temp
+
+            high--
+        }
+    }
+
+	return nums
+}

medium/75_sort_colors/75_sort_colors_test.go

@@ -0,0 +1,34 @@
+package sort_colors
+
+import (
+	"reflect"
+	"testing"
+)
+
+func TestSortColors(t *testing.T) {
+	tests := []struct {
+		name	string
+		nums	[]int 
+		expectedResult []int
+	} {
+		{
+			name: "Test Case 1",
+			nums: []int{2,0,2,1,1,0},
+			expectedResult: []int{0,0,1,1,2,2},
+		}, 
+		{
+			name: "Test Case 2",
+			nums: []int{2, 0 ,1},
+			expectedResult: []int{0, 1, 2},
+		}, 
+	}
+
+	for _, test := range tests {
+		t.Run(test.name, func(t * testing.T) {
+			got := SortColors(test.nums)
+			if  !reflect.DeepEqual(got, test.expectedResult){
+				t.Errorf("SortColors(%v) = %v; want = %v", test.nums, got, test.expectedResult)
+			}
+		})
+	}
+}