Add: Split a String in Balanced Strings

Author: openset

problems/split-a-string-in-balanced-strings/split_a_string_in_balanced_strings.go

@@ -0,0 +1,16 @@
+package problem_1221
+
+func balancedStringSplit(s string) int {
+	ans, flag := 0, 0
+	for _, b := range s {
+		if b == 'L' {
+			flag--
+		} else {
+			flag++
+		}
+		if flag == 0 {
+			ans++
+		}
+	}
+	return ans
+}

problems/split-a-string-in-balanced-strings/split_a_string_in_balanced_strings_test.go

@@ -0,0 +1,31 @@
+package problem_1221
+
+import "testing"
+
+type caseType struct {
+	input    string
+	expected int
+}
+
+func TestBalancedStringSplit(t *testing.T) {
+	tests := [...]caseType{
+		{
+			input:    "RLRRLLRLRL",
+			expected: 4,
+		},
+		{
+			input:    "RLLLLRRRLR",
+			expected: 3,
+		},
+		{
+			input:    "LLLLRRRR",
+			expected: 1,
+		},
+	}
+	for _, tc := range tests {
+		output := balancedStringSplit(tc.input)
+		if output != tc.expected {
+			t.Fatalf("input: %v, output: %v, expected: %v", tc.input, output, tc.expected)
+		}
+	}
+}