Leetcode每日刷题【中】--Day 13

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大

Leetcode每日刷题【中】--Day 13
注:A,B,D是前缀和结果的矩阵中的值,C’是原矩阵,C与欲求的值

上一篇:leetcode1738. 找出第 K 大的异或坐标值


下一篇:【力扣】1442. 形成两个异或相等数组的三元组数目