LeetCode Maximal Square

LeetCode Maximal Square

Given a 2D binary matrix filled with 0's and 1's, find the largest square containing all 1's and return its area.

For example, given the following matrix:

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

Return 4.

题意是给定一个二维字符矩阵，只有0和1字符，要求找出只包含字符1的最大子矩阵，返回其大小。

解题思路：动态规划（DP）；DP问题的关键点在于 寻找状态方程，本题我们把状态定义为在（i，j）位置处能达到的只包含字符1的最大矩阵边的长度。这样，面积就是长度的平方。现在关键是计算dp[i][j]。

需要根据matrix矩阵的元素来计算dp。首先需要找到起始位置：由matrix的第一行和第一列计算出dp的第一行和第一列的值，即 dp[i][j] = matrix[i][j] - '0'（其中i=0或j=0）。然后，当i > 0 且 j > 0时，如果matrix[i][j] == '0'那么dp[i][j] = 0（因为题目要求不能包含有‘0’的矩阵）；当matrix[i][j] == '1'时，该如何计算dp[i][j]？我们将dp此时的情况写出来：

. . . . . .

. dp[i-1][j-1] dp[i-1][j]

dp[i][j-1] dp[i][j]

. . . . . . .

需要根据dp[i-1][j-1]，dp[i-1][j]，dp[i][j-1]来计算dp[i][j]。因为其它三个已经在之前计算好了，dp[i][j] = 三者最小值 + 1,；于是

1、i = 0或j = 0时，dp[i][j] = matrix[i][j] - '0'

2、i>0 且 j > 0时，如果matr[i][j] = ‘0’，则dp[i][j]=0；如果matrix[i][j]=‘1’，则dp[i][j] = min(dp[i-1][j-1], dp[i-1][j], dp[i][j-1]) + 1

int row = matrix.size();
	if (row == 0)
		return 0;
	int col = matrix[0].size();	

	int maximal = 0;
	vector<vector<int> > dp(row, vector<int> (col, 0));//生成对应于matrix的矩阵

	//根据matrix初始化dp第一列
	for (int i = 0; i < row; i++)
	{
		dp[i][0] = matrix[i][0] - '0';
				maximal = max(maximal, dp[i][0]);
	} 
	//根据matrix初始化dp第一行
	for (int j = 1; j < col; j++)
	{
		dp[0][j] = matrix[0][j]- '0';		
		maximal = max(maximal, dp[0][j]);
	}

	for (int i = 1; i < row; i++)
	{
		for (int j = 1; j < col; j++)
		{
			if (matrix[i][j] == '1')
			{
				dp[i][j] = min(dp[i-1][j], min(dp[i - 1][j - 1], dp[i][j - 1])) + 1;
				maximal = max(maximal, dp[i][j]);
			}
			
		}
	}

	return maximal * maximal;

当然，以上使用了dp矩阵来根据matrix的值来赋值，但其实每次计算的时候并没有使用整个dp矩阵，而只是用了dp[i-1][j-1]，dp[i][j-1]，dp[i-1][j]，因此可以使用两个向量来保存需要的信息，而不是使用整个矩阵：

int row = matrix.size();
	if (row == 0)
		return 0;
	int col = matrix[0].size();

	int maximal = 0;

	vector<int> pre_col(row, 0);//前一列存储的元素
	vector<int> cur_col(row, 0);//当前列存储的元素

	for (int i = 0; i < row; i++)
	{
		pre_col[i] = matrix[i][0] - '0';
		maximal = max(maximal, pre_col[i]);
	}
		
	for (int j = 1; j < col; j++)
	{
		cur_col[0] = matrix[0][j] - '0';
		maximal = max(maximal, cur_col[0]);
		for (int i = 1; i < row; i++)
		{
			if (matrix[i][j] == '1')
			{
				cur_col[i] = min(cur_col[i - 1], min(pre_col[i - 1], pre_col[i])) + 1;
				maximal = max(maximal, cur_col[i]);
			}
		}

		cur_col = pre_col;
		fill(cur_col.begin(), cur_col.end(), 0);
	}

	return maximal * maximal;

当然，也可以就地使用matrix矩阵、而不是用其他辅助空间，但是此时会改动matrix，倘若赋值时超过了char的范围，则越界，程序安全性减低，因此最好不在原来的matrix上做改动。