Merge pull request #642 from openset/develop

Add: 2 Keys Keyboard

Author: Shuo

problems/2-keys-keyboard/2_keys_keyboard.go

@@ -1 +1,13 @@
 package p_2_keys_keyboard
+
+func minSteps(n int) int {
+	if n == 1 {
+		return 0
+	}
+	for i := 2; i < n; i++ {
+		if n%i == 0 {
+			return i + minSteps(n/i)
+		}
+	}
+	return n
+}

problems/2-keys-keyboard/2_keys_keyboard_test.go

@@ -1 +1,35 @@
 package p_2_keys_keyboard
+
+import "testing"
+
+type caseType struct {
+	input    int
+	expected int
+}
+
+func TestMinSteps(t *testing.T) {
+	tests := [...]caseType{
+		{
+			input:    1,
+			expected: 0,
+		},
+		{
+			input:    3,
+			expected: 3,
+		},
+		{
+			input:    30,
+			expected: 10,
+		},
+		{
+			input:    97,
+			expected: 97,
+		},
+	}
+	for _, tc := range tests {
+		output := minSteps(tc.input)
+		if output != tc.expected {
+			t.Fatalf("input: %v, output: %v, expected: %v", tc.input, output, tc.expected)
+		}
+	}
+}