小美拿到了一个 n*n 的矩阵,其中每个元素是 0 或者 1。 小美认为一个矩形区域是完美的,当且仅当该区域内 0 的数量恰好等于 1 的数量。 现在,小美希望你回答有多少个 i*i 的完美矩形区域。你需要回答 1≤ i ≤ n 的所有答案。
第一行输入一个正整数 n ,代表矩阵大小。 接下来的 n 行,每行输入一个长度为 n 的 01 串,用来表示矩阵。 1≤ n ≤ 200
输出 n 行,第 i 行输出 i*i 的完美矩形区域的数量。
4 1010 0101 1100 0011
0 7 0 1
考点:前缀和
数据规模 n ≤ 200 | 限制 1 秒 / 256MB | 标准输入输出
推荐方向:前缀和
本题切入点
本题要枚举所有 i×i 子矩阵并判断 0/1 数量相等,先用二维前缀和把每个子矩阵的和降到 O(1),再对每个边长 i 枚举左上角,整体 O(n³) 在 n≤200 下可行。
预处理前缀和数组,把「区间求和」从 O(n) 降到 O(1)。
思路框架(前缀和 通法 · 非本题专属)
实现要点:开 long long,避免 n 较大时求和溢出。
复杂度:时间 预处理 O(n),每次查询 O(1) | 空间 O(n)
该范式的通法易错点
对照本题
样例 1
4 / 1010 / 0101 / 1100 / 00110 / 7 / 0 / 1解析由校招宝本地引擎整理(依据源站考点标签 / 人工判题标注 / 题面规模信号),非官方题解,仅供思路参考。
本题来源:2024年春招-美团-前端移动端-第一批笔试;2024年春招-美团-算法策略-第一批笔试;2024年春招-美团-技术岗-第一批笔试 等。