update

oj/google-2019/candy.go

@@ -73,6 +73,12 @@ func maxInt(x, y int) int {
 }
 
 func main() {
+	// Case 1
+	// 1 1 0 0 0
+	// 0 1 1 0 0
+	// 0 0 1 1 0
+	// 0 0 0 1 1
+	// 0 0 0 0 1
 	var candies []candy
 	for i := 0; i < 5; i++ {
 		candies = append(candies, candy{i, i})
@@ -80,5 +86,18 @@ func main() {
 			candies = append(candies, candy{i, i + 1})
 		}
 	}
-	fmt.Println(getMinimalTime(candies, 5))
+	fmt.Println(getMinimalTime(candies, 5) == 8)
+	// Case 2
+	// Fill all grid with candies
+	candies = make([]candy, 0)
+	for i := 0; i < 5; i++ {
+		for j := 0; j < 5; j++ {
+			candies = append(candies, candy{i, j})
+		}
+	}
+	// Case 3
+	// No candy
+	fmt.Println(getMinimalTime(candies, 5) == 24)
+	candies = make([]candy, 0)
+	fmt.Println(getMinimalTime(candies, 5) == 4)
 }