A: String Matching in an Array

Author: Shuo

problems/string-matching-in-an-array/string_matching_in_an_array.go

@@ -0,0 +1,21 @@
+package problem1408
+
+import "strings"
+
+func stringMatching(words []string) []string {
+	var ans []string
+	ok := func(str string) bool {
+		for _, word := range words {
+			if word != str && strings.Contains(word, str) {
+				return true
+			}
+		}
+		return false
+	}
+	for _, word := range words {
+		if ok(word) {
+			ans = append(ans, word)
+		}
+	}
+	return ans
+}

problems/string-matching-in-an-array/string_matching_in_an_array_test.go

@@ -0,0 +1,34 @@
+package problem1408
+
+import (
+	"reflect"
+	"testing"
+)
+
+type testType struct {
+	in   []string
+	want []string
+}
+
+func TestStringMatching(t *testing.T) {
+	tests := [...]testType{
+		{
+			in:   []string{"mass", "as", "hero", "superhero"},
+			want: []string{"as", "hero"},
+		},
+		{
+			in:   []string{"leetcode", "et", "code"},
+			want: []string{"et", "code"},
+		},
+		{
+			in:   []string{"blue", "green", "bu"},
+			want: nil,
+		},
+	}
+	for _, tt := range tests {
+		got := stringMatching(tt.in)
+		if !reflect.DeepEqual(got, tt.want) {
+			t.Fatalf("in: %v, got: %v, want: %v", tt.in, got, tt.want)
+		}
+	}
+}