A: Maximum Score After Splitting a String

Author: Shuo

problems/maximum-score-after-splitting-a-string/maximum_score_after_splitting_a_string.go

@@ -0,0 +1,20 @@
+package problem1422
+
+import "strings"
+
+func maxScore(s string) int {
+	ans := 0
+	n0 := 0
+	n1 := strings.Count(s, "1")
+	for _, v := range s[:len(s)-1] {
+		if v == '1' {
+			n1--
+		} else {
+			n0++
+		}
+		if ans < n0+n1 {
+			ans = n0 + n1
+		}
+	}
+	return ans
+}

problems/maximum-score-after-splitting-a-string/maximum_score_after_splitting_a_string_test.go

@@ -0,0 +1,35 @@
+package problem1422
+
+import "testing"
+
+type testType struct {
+	in   string
+	want int
+}
+
+func TestMaxScore(t *testing.T) {
+	tests := [...]testType{
+		{
+			in:   "011101",
+			want: 5,
+		},
+		{
+			in:   "00111",
+			want: 5,
+		},
+		{
+			in:   "1111",
+			want: 3,
+		},
+		{
+			in:   "000000",
+			want: 5,
+		},
+	}
+	for _, tt := range tests {
+		got := maxScore(tt.in)
+		if got != tt.want {
+			t.Fatalf("in: %v, got: %v, want: %v", tt.in, got, tt.want)
+		}
+	}
+}