酷代码 AI
菜单
全部AI  /  热门  /  AI写代码神器
服务商

描述 给定一个 n × n 的矩阵,在其上进行若干次操作。每次操作,你可以将一个 k × k 的连续子矩阵里的所有数全部加上1或者全都减去1。 初始时,矩阵中有 m 个位置上的数不为 0 ,其他位置上的数均为 0。请问至少需要多少次操作,可以将矩阵中的所有数字都变为 0。 格式 输入 第一行三个整数 n, m, k 分别表示矩阵大小、非零格数和每次修改的连续子矩阵大小。 接下来 m 行,每行三个整数 x, y, z 表示初始时矩阵的第 x 行第 y列上的数为z 。 输出 一行,一个整数,表示最少操作次数。特别地,如果无法使矩阵中所有数都变为0,输出-1.

以下是使用 Python 解决此问题的示例代码: ```python import math def check_possible(n, m, k, positions): if k > n: return -1 count = 0 matrix = [[0] * n for _ in range(n)] for x, y, z in positions: matrix[x - 1][y - 1] = z def operate(x, y, add): for i in range(x, min(x + k, n)): for j in range(y, min(y + k, n)): matrix[i][j] += add while True: found_non_zero = False for i in range(n): for j in range(n): if matrix[i][j]!= 0: found_non_zero = True if i + k <= n and j + k <= n: operate(i, j, -matrix[i][j] // abs(matrix[i][j])) count += 1 if not found_non_zero: break for row in matrix: for num in row: if num!= 0: return -1 return count n, m, k = map(int, input().split()) positions = [list(map(int, input().split())) for _ in range(m)] print(check_possible(n, m, k, positions)) ``` 这段代码首先构建初始矩阵,然后通过不断的操作尝试将所有元素变为 0 ,如果最终无法全部变为 0 则返回 -1 ,否则返回操作次数。 [2024-10-24 14:05:14 | AI写代码神器 | 360点数解答]

相关提问