1738. 找出第 K 大的异或坐标值

Posted 2021-05-23 lgz0921

题目链接：https://leetcode-cn.com/problems/find-kth-largest-xor-coordinate-value/

思路：从左上角往右下角一个一个遍历即可，统计前缀异或和，放在一个list中，最后排序返回第k大的即可。

一个帮助理解的图：

每次计算右下角的异或和的时候result[i - 1][j - 1]这块矩形区域计算了两次，所以需要最后再xor一次。

公式：result[i][j] = matrix[i][j] xor result[i - 1][j] xor result[i][j - 1] xor result[i - 1][j - 1]     其他的就是边界处理上的细节了，这里不做说明了（很简单~~~）。

上代码：

class Solution {
    fun kthLargestValue(matrix: Array<IntArray>, k: Int): Int {
        val result = Array(size = matrix.size, init = { IntArray(matrix[0].size) })
        val resultList = ArrayList<Int>()
        for (i in matrix.indices) {
            for (j in matrix[0].indices) {
                when {
                    i == 0 && j == 0 -> result[i][j] = matrix[i][j]
                    i == 0 -> result[i][j] = matrix[i][j] xor result[i][j - 1]
                    j == 0 -> result[i][j] = matrix[i][j] xor result[i - 1][j]
                    else -> result[i][j] = matrix[i][j] xor result[i - 1][j] xor result[i][j - 1] xor result[i - 1][j - 1]
                }
                resultList.add(result[i][j])
            }
        }
        resultList.sort()
        return resultList[resultList.size - 1 - (k - 1)]
    }
}