Add: Assign Cookies

Author: openset

problems/assign-cookies/assign_cookies.go

@@ -1 +1,16 @@
-package assign_cookies
+package problem_455
+
+import "sort"
+
+func findContentChildren(g []int, s []int) int {
+	sort.Ints(g)
+	sort.Ints(s)
+	var ans, i, j int
+	for i < len(g) && j < len(s) {
+		if g[i] <= s[j] {
+			ans, i = ans+1, i+1
+		}
+		j++
+	}
+	return ans
+}

problems/assign-cookies/assign_cookies_test.go

@@ -1 +1,30 @@
-package assign_cookies
+package problem_455
+
+import "testing"
+
+type caseType struct {
+	g        []int
+	s        []int
+	expected int
+}
+
+func TestFindContentChildren(t *testing.T) {
+	tests := [...]caseType{
+		{
+			g:        []int{1, 2, 3},
+			s:        []int{1, 1},
+			expected: 1,
+		},
+		{
+			g:        []int{1, 2},
+			s:        []int{1, 2, 3},
+			expected: 2,
+		},
+	}
+	for _, tc := range tests {
+		output := findContentChildren(tc.g, tc.s)
+		if output != tc.expected {
+			t.Fatalf("input: %v, output: %v, expected: %v", tc.g, output, tc.expected)
+		}
+	}
+}