在Python中查找第k个最大异或坐标值的程序
假设我们有一个mxn矩阵。和另一个值k。这里矩阵的坐标(a,b)的值是所有矩阵[i,j]的异或,其中i在范围(0到a)和j在范围(0到b)。我们必须找到矩阵所有坐标的第k个最大值(1-indexed)。
所以,如果输入是这样的
并且k=1,那么输出将是7,因为坐标(0,1)的值是5XOR2=7,这是最大的
示例
让我们看看以下实现以获得更好的理解-
def solve(matrix, k): m, n = len(matrix), len(matrix[0]) for i in range(m): for j in range(n): if j: matrix[i][j] ^= matrix[i][j-1] seen = {} count = 0 for i in range(n): for j in range(m): if j: matrix[j][i] ^= matrix[j-1][i] seen[matrix[j][i]] = seen.get(matrix[j][i], 0) + 1 count += 1 if count > k: min_value = min(seen) seen[min_value] -= 1 if not seen[min_value]: seen.pop(min_value) return min(seen) matrix = [[5,2],[1,6]] k = 1 print(solve(matrix, k))
输入
[[5,2],[1,6]], 1输出结果
7