221 wrong answer

Author: aQua

Algorithms/0221.maximal-square/maximal-square.go

@@ -70,8 +70,8 @@ func maximalSquare(matrix [][]byte) int {
 				}
 			}
 			temp++
-			if max < temp*temp {
-				max = temp * temp
+			if max < temp {
+				max = temp
 			}
 
 			if a+temp >= m || b+temp >= n {
@@ -87,5 +87,5 @@ func maximalSquare(matrix [][]byte) int {
 		}
 	}
 
-	return max
+	return max * max
 }

Algorithms/0221.maximal-square/maximal-square_test.go

@@ -13,6 +13,16 @@ var tcs = []struct {
 	ans    int
 }{
 
+	{
+		[][]byte{
+			[]byte("11111111"),
+			[]byte("11111110"),
+			[]byte("11111110"),
+			[]byte("11111000"),
+			[]byte("01111000"),
+		},
+		16,
+	},
 	{
 		[][]byte{
 			[]byte("0001"),