1738. 找出第 K 大的异或坐标值
又是前缀和,连续几次被调教到了,这次终于翻身一次了!似乎比官方部分题解要节约用时和节省空间的
思路还是像之前的一样,前缀异或,然后在python找出第k大那是易如饮水的事了。但是重点要理解一句话
矩阵中坐标 (a, b) 的 值 由对所有满足 0 <= i <= a < m 且 0 <= j <= b < n 的元素 matrix[i][j](下标从 0 开始计数)执行异或运算得到
这里画个图之后,就能够很容易理解题意了。图片在下方~
class Solution:
def kthLargestValue(self, matrix: List[List[int]], k: int) -> int:
m, n = len(matrix), len(matrix[0])
xor_matrix = [[0]*n for i in range(m)] # 前缀异或的结果矩阵
xor_matrix[0][0] = matrix[0][0] # 第一个数应要求是原矩阵第一个数
for x in range(m): # 行遍历
for y in range(n): # 列遍历
if x == 0 and y >= 1: # 如果是第一行,只需要考虑一行就行了
xor_matrix[x][y] = matrix[x][y] ^ xor_matrix[x][y-1]
if y == 0 and x >= 1: # 是第一列,同上
xor_matrix[x][y] = matrix[x][y] ^ xor_matrix[x-1][y]
if x > 0 and y > 0: # 在中间部分,如下图所示
xor_matrix[x][y] = xor_matrix[x-1][y] ^ xor_matrix[x][y-1] ^ xor_matrix[x-1][y-1] ^ matrix[x][y]
xor_matrix = [x for i in xor_matrix for x in i] # 展开
xor_matrix.sort() # 排序方便查找第K大
return xor_matrix[-k] # 第K大
注:A,B,D是前缀和结果的矩阵中的值,C’是原矩阵,C与欲求的值