小美的平衡矩阵【Python实现】

B站影视 日本电影 2025-05-31 12:58 1

摘要:n = int(input)matrix = [input.strip for _ in range(n)]# 构建前缀和矩阵prefix = [[0] * (n + 1) for _ in range(n + 1)]for i in range(1, n +

n = int(input)matrix = [input.strip for _ in range(n)]# 构建前缀和矩阵prefix = [[0] * (n + 1) for _ in range(n + 1)]for i in range(1, n + 1): for j in range(1, n + 1): prefix[i][j] = prefix[i-1][j] + prefix[i][j-1] - prefix[i-1][j-1] + int(matrix[i-1][j-1])# 统计每个k的完美矩形数量result = [0] * nfor k in range(1, n + 1): count = 0 for i in range(n - k + 1): for j in range(n - k + 1): # 计算子矩阵中1的个数 total_ones = prefix[i + k][j + k] - prefix[i][j + k] - prefix[i + k][j] + prefix[i][j] if total_ones * 2 == k * k: count += 1 result[k - 1] = count# 输出结果for num in result: print(num)

来源:小橘喵科技

相关推荐