最大子方阵

题目

有一个方阵，其中每个单元(像素)非黑即白(非0即1)，请设计一个高效算法，找到四条边颜色相同的最大子方阵。
给定一个01方阵mat，同时给定方阵的边长n，请返回最大子方阵的边长。保证方阵边长小于等于100。

测试样例：
[[1,1,1],[1,0,1],[1,1,1]],3
返回：3

解答

这题的思路主要是对矩阵从大到小进行扫描，直至扫描出符合条件的方阵，那么很显然最优解就是方阵自己本身，在这里我使用了四层循环去解决（感觉不好，时间复杂度太大），第一层循环用于确定最大子方阵的边长，第二第三层用于确定扫描的起点，第四层用于扫描，扫描前首先对边界条件进行判断，判断通过后同时对四条边进行扫描，一旦发现和起始点颜色不同，立刻跳出循环，转向第二个起点。
如果第四层循环结束后，仍未跳出，则证明符合条件，输出边长。若四层循环结束后，仍未输出，则证明最大子方阵为1，此为最差解，则输出1.
总体来说这题不难，主要是边界条件的设定会造成一些麻烦，还有用了四层循环，显得有些美中不足。
具体代码实现如下：

import java.util.*;

public class SubMatrix {
    public int maxSubMatrix(int[][] mat, int n) {
        // write code here
        for(int count = n - 1;count > 0;count--){
            for(int i = 0;i < n;i++){
                for(int j = 0;j < n;j++){
                    for(int k = 0;k < count + 1;k++){
                        if(i + count < n&&j + count < n&&i + k < n&&j + k < n){
                            if(mat[i][j + k] != mat[i][j]){
                                break;
                            }
                            if(mat[i + k][j] != mat[i][j]){
                                break;
                            }
                            if(mat[i + count][j + k] != mat[i][j]){
                                break;
                            }
                            if(mat[i + k][j + count] != mat[i][j]){
                                break;
                            }
                            if(k == count){
                                return count + 1;
                            }                            
                        }
                    }                    
                }
            }
        }
        return 1;
        }
    }