T2 小美的平衡矩阵(25分) - 美团编程题 & 题解

考试平台: 牛客网

题目类型: 30道单选题(60分)+ 2 道编程题 (15分 + 25分)

考试时间: 2024-03-09 (两小时)

题目描述

小美拿到了一个n*n的矩阵,其中每个元素是 0 或者 1。

小美认为一个矩形区域是完美的,当且仅当该区域内 0 的数量恰好等于 1 的数量。

现在,小美希望你回答有多少个i*i的完美矩形区域。你需要回答 1 ≤ i ≤ n 1\leq i \leq n 1≤i≤n的所有答案。

输入描述

第一行输入一个正整数n ,代表矩阵大小。

接下来的n行,每行输入一个长度为n的 01 串,用来表示矩阵。
1 ≤ n ≤ 200 1\leq n \leq 200 1≤n≤200

输出描述

输出n行,第i行输出 i*i的完美矩形区域的数量。

示例

输入:
4
1010
0101
1100
0011

输出:
0
7
0
1

题解

暴力枚举所有情况, 在暴力枚举的情况下使用前缀和优化检验 "是否是完美矩形区域" 的方法。

python 复制代码
from collections import defaultdict

n = int(input())

grid = [input() for _ in range(n)]

# 前缀和 psum[r][i] 表示 grid[r][0:i] 中 "1" 的个数
psum = [[0] * (n+1) for _ in range(n)]
for r in range(n):
    for i, v in enumerate(grid[r]):
        if v == "1":
            psum[r][i+1] = psum[r][i] + 1
        else:
            psum[r][i+1] = psum[r][i]
        

def check(r, c, w) -> bool:
    """
    检验左上角为 (r,c) 宽度为 w 的是否是完美矩形区域
    """
    global n, grid, psum
    if r + w > n or c + w > n:
        return False

    # 统计矩形区域内 1 的数量
    cnt = 0
    for rc in range(w):
        cnt += psum[r + rc][c + w] - psum[r + rc][c]
	
    # 如果 1 的数据和0的数量相同则,  cnt * 2 == w * w
    return cnt * 2 == w * w


cnt = defaultdict(dict)

# 暴力枚举所有的情况
for r in range(n):
    for c in range(n):
        for w in range(2, n + 1): # 枚举举行宽度
            if check(r, c, w):
                cnt[w] = cnt.get(w, 0) + 1

# 打印结果输出
for i in range(1, n + 1):
    print(cnt.get(i, 0))
相关推荐
秃头佛爷4 分钟前
Python学习大纲总结及注意事项
开发语言·python·学习
浮生如梦_1 小时前
Halcon基于laws纹理特征的SVM分类
图像处理·人工智能·算法·支持向量机·计算机视觉·分类·视觉检测
深度学习lover1 小时前
<项目代码>YOLOv8 苹果腐烂识别<目标检测>
人工智能·python·yolo·目标检测·计算机视觉·苹果腐烂识别
API快乐传递者2 小时前
淘宝反爬虫机制的主要手段有哪些?
爬虫·python
励志成为嵌入式工程师3 小时前
c语言简单编程练习9
c语言·开发语言·算法·vim
捕鲸叉4 小时前
创建线程时传递参数给线程
开发语言·c++·算法
A charmer4 小时前
【C++】vector 类深度解析:探索动态数组的奥秘
开发语言·c++·算法
wheeldown4 小时前
【数据结构】选择排序
数据结构·算法·排序算法
阡之尘埃4 小时前
Python数据分析案例61——信贷风控评分卡模型(A卡)(scorecardpy 全面解析)
人工智能·python·机器学习·数据分析·智能风控·信贷风控
观音山保我别报错5 小时前
C语言扫雷小游戏
c语言·开发语言·算法