有一个方阵,其中每个单元(像素)非黑即白(非0即1),请设计一个高效算法,找到四条边颜色相同的最大子方阵。
给定一个01方阵mat,同时给定方阵的边长n,请返回最大子方阵的边长。保证方阵边长小于等于100。
测试样例:
[[1,1,1],[1,0,1],[1,1,1]],3
返回:3
class SubMatrix { public: int maxSubMatrix(vector<vector<int> > mat, int n) { // write code here //选定最大子方阵的边长,选择顶点,判断是合法 int maxlength=n; while(maxlength) //选择边长 { for(int i=0;i<=n-maxlength;i++) { for(int j=0;j<=n-maxlength;j++) { //当前顶点为[i,j]; int key=mat[i][j]; //判断是否合法 bool flag=true; for(int k=0;k<maxlength;k++) { int k1=mat[i][j+k]; int k2=mat[i+maxlength-1][j+k]; int k3=mat[i+k][j]; int k4=mat[i+k][j+maxlength-1]; if(k1!=key||k2!=key||k3!=key||k4!=key) { flag=false; break; } } if(flag) return maxlength; } } maxlength--; } return 0; } };