Add: Bulb Switcher

Author: openset

internal/leetcode/problems_status.go

@@ -94,6 +94,7 @@ var problemStatus = map[int]bool{
 	290:  true,
 	292:  true,
 	303:  true,
+	319:  true,
 	326:  true,
 	342:  true,
 	344:  true,

problems/bulb-switcher/bulb_switcher.go

@@ -1 +1,7 @@
-package bulb_switcher
+package problem_319
+
+import "math"
+
+func bulbSwitch(n int) int {
+	return int(math.Sqrt(float64(n)))
+}

problems/bulb-switcher/bulb_switcher_test.go

@@ -1 +1,31 @@
-package bulb_switcher
+package problem_319
+
+import "testing"
+
+type caseType struct {
+	input    int
+	expected int
+}
+
+func TestBulbSwitch(t *testing.T) {
+	tests := [...]caseType{
+		{
+			input:    3,
+			expected: 1,
+		},
+		{
+			input:    4,
+			expected: 2,
+		},
+		{
+			input:    5,
+			expected: 2,
+		},
+	}
+	for _, tc := range tests {
+		output := bulbSwitch(tc.input)
+		if output != tc.expected {
+			t.Fatalf("input: %v, output: %v, expected: %v", tc.input, output, tc.expected)
+		}
+	}
+}