Merge pull request 6boris#138 from kylesliu/develop

Kyle Liu · web-flow · commit ded32f94c89d · 2020-04-17T23:43:23.000+08:00
add 643 solution
diff --git a/src/0643.Maximum-Average-Subarray-I/Solution.go b/src/0643.Maximum-Average-Subarray-I/Solution.go
@@ -1,5 +1,21 @@
 package Solution
 
-func Solution(x bool) bool {
-	return x
+func findMaxAverage(nums []int, k int) float64 {
+	tmp, ans := 0, -1
+	for i := 0; i < len(nums); i++ {
+		tmp += nums[i]
+		if i >= k-1 {
+			ans = max(ans, tmp)
+			tmp -= nums[i-k+1]
+		}
+	}
+
+	return float64(ans) / float64(k)
+}
+
+func max(x, y int) int {
+	if x > y {
+		return x
+	}
+	return y
 }
diff --git a/src/0643.Maximum-Average-Subarray-I/Solution_test.go b/src/0643.Maximum-Average-Subarray-I/Solution_test.go
@@ -10,21 +10,20 @@ func TestSolution(t *testing.T) {
 	//	测试用例
 	cases := []struct {
 		name   string
-		inputs bool
-		expect bool
+		input1 []int
+		input2 int
+		expect float64
 	}{
-		{"TestCase", true, true},
-		{"TestCase", true, true},
-		{"TestCase", false, false},
+		{"TestCase", []int{1, 12, -5, -6, 50, 3}, 4, 12.75},
 	}
 
 	//	开始测试
 	for i, c := range cases {
 		t.Run(c.name+" "+strconv.Itoa(i), func(t *testing.T) {
-			got := Solution(c.inputs)
+			got := findMaxAverage(c.input1, c.input2)
 			if !reflect.DeepEqual(got, c.expect) {
 				t.Fatalf("expected: %v, but got: %v, with inputs: %v",
-					c.expect, got, c.inputs)
+					c.expect, got, c.input1)
 			}
 		})
 	}
