Add: Maximize Distance to Closest Person

Author: openset

problems/maximize-distance-to-closest-person/maximize_distance_to_closest_person.go

@@ -1 +1,28 @@
 package maximize_distance_to_closest_person
+
+func maxDistToClosest(seats []int) int {
+	ans, prev, future, l := 0, -1, 0, len(seats)
+	for i, v := range seats {
+		if v == 1 {
+			prev = i
+		} else {
+			for future < l && seats[future] == 0 || future < i {
+				future++
+			}
+			left, right := l, l
+			if prev != -1 {
+				left = i - prev
+			}
+			if future != l {
+				right = future - i
+			}
+			if right < left {
+				left = right
+			}
+			if left > ans {
+				ans = left
+			}
+		}
+	}
+	return ans
+}

problems/maximize-distance-to-closest-person/maximize_distance_to_closest_person_test.go

@@ -1 +1,32 @@
 package maximize_distance_to_closest_person
+
+import "testing"
+
+type caseType struct {
+	input    []int
+	expected int
+}
+
+func TestMaxDistToClosest(t *testing.T) {
+	tests := [...]caseType{
+		{
+			input:    []int{1, 0, 0, 0, 1, 0, 1},
+			expected: 2,
+		},
+		{
+			input:    []int{1, 0, 0, 0},
+			expected: 3,
+		},
+
+		{
+			input:    []int{0, 0, 0, 1},
+			expected: 3,
+		},
+	}
+	for _, tc := range tests {
+		output := maxDistToClosest(tc.input)
+		if output != tc.expected {
+			t.Fatalf("input: %v, output: %v, expected: %v", tc.input, output, tc.expected)
+		}
+	}
+}