leetcode 221.最大正方形 golang实现

描述

在一个由 0 和 1 组成的二维矩阵内，找到只包含 1 的最大正方形，并返回其面积。

示例:

输入: 

1 0 1 0 0
1 0 1 1 1
1 1 1 1 1
1 0 0 1 0

输出: 4

思路

动态规划问题

我们用一个例子来解释这个方法：

0 1 1 1 0
1 1 1 1 1
0 1 1 1 1
0 1 1 1 1
0 0 1 1 1

dp(i,j) 表示的是由 1 组成的最大正方形的边长
当 nums[i][j] = 1
dp[i][j] = min(dp(i−1, j), dp(i−1, j−1), dp(i, j−1))+1
当 nums[i][j] = 0
dp[i][j] = 0

dp[i][j]=min(dp(i−1, j), dp(i−1, j−1), dp(i, j−1))+1的意思是  如果nums[i][j] = 1 则 dp[i][j]表示的边长 = min(左， 左上， 上) 的边长的最小值 + 1

可以看这里leetcode的图解 更容易理解

实现

func maximalSquare(matrix [][]byte) int {
	rows := len(matrix)
    if rows == 0{
		return 0
	}
	cols := len(matrix[0])
	maxSideLen := 0

	dp := make([][]int, rows + 1)
	dp[0] = make([]int, cols + 1)

	for i := 1; i <= rows; i++{
		dp[i] = make([]int, cols + 1)
		for j := 1; j <= cols; j++{
			if matrix[i-1][j-1] == '0'{
				dp[i][j] = 0
			}else{
				dp[i][j] = MinThree(dp[i-1][j], dp[i][j-1], dp[i-1][j-1]) + 1
				if dp[i][j] > maxSideLen{
					maxSideLen = dp[i][j]
				}
			}
		}
	}

	return maxSideLen*maxSideLen
}

func MinThree(a, b, c int) int {
	var min int
	if a >= b {
		min = b
	} else {
		min = a
	}

	if min >= c{
		return c
	}else{
		return min
	}
}