手机版 欢迎访问it开发者社区(www.mfbz.cn)网站

当前位置: > 开发

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

时间:2021/5/19 15:38:02|来源:|点击: 次

题目链接: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)]
    }
}

 

Copyright © 2002-2019 某某自媒体运营 版权所有