Tsai笔记：C++学习随性笔记（5）—— 华为面试题目总结

Tsai笔记：C++学习随性笔记（5）—— 华为面试题目总结

一.在一个二维01矩阵中找到全为1的最大正方形，返回其面积。
如以下样例样例:

返回 4(2*2)

解题思想

1. 先构造一个辅助矩阵tmp，把matrix矩阵的第一行和第一列元素直接复制；

2. 对于剩下元素，若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;
}