Tsai笔记:C++学习随性笔记(5)—— 华为面试题目总结
一.在一个二维01矩阵中找到全为1的最大正方形,返回其面积。
如以下样例样例:
返回 4(2*2)
解题思想
-
先构造一个辅助矩阵tmp,把matrix矩阵的第一行和第一列元素直接复制;
-
对于剩下元素,若matrix[i][j]=0, 则tmp[i][j]=0;若matrix[i][j]=1,则tmp[i][j]= min(tmp[i-1][j],tmp[i-1][j-1],tmp[i][j-1]) +1;
tmp[i][j]的值表示的是矩阵中以(i,j)位置为右下角的正方形的边长;
#include <iostream>
#include <vector>
int MaxSquare(vector<vector<int>>& matrix){
int i,j,rows,cols,tmp = 0;
// 1.check
rows = matrix.size();
if(!rows)
return 0;
cols = matrix[0].size();
if(!cols)
return 0;
//2.辅助矩阵初始化为0
vector<vector<int>> f(rows,vector<int>(cols,0));
//3.复制首行首列
for(i = 0; i < rows;i++){
if(matrix[i][0] == 1){
f[i][0] = 1;
tmp = 1;
}
}
for(i = 0;i < cols; i++){
if(matrix[0][i] == 1){
f[0][i] = 1;
tmp = 1;
}
}
//4.去除最大矩阵维数
for(i = 1;i < rows; i++){
for(j = 1;j < cols; j++){
if(matrix[i][j] == 0)
continue;
f[i][j] = min(min(matrix[i-1][j],matrix[i-1][j-1]),matrix[i][j])+1;
if(f[i][j] > tmp)
tmp = f[i][j];
}
}
return tmp*tmp;
}
int main() {
vector<vector<int>> matrix = {{1, 0, 1, 0, 0},
{1, 0, 1, 1, 1},
{1, 1, 1, 1, 1},
{1, 0, 0, 1, 0}};
int a = MaxSquare(matrix);
cout << "nums : " << a << endl;
}