leetcode-1738-找出第k大的异或坐标值

Posted 2021-05-20 caoshikui

leetcode-1738-找出第k大的异或坐标值

描述

思路

1. 采用二维前缀和的方式
2. pre[m+1][n+1]数组记录前缀和
3. pre[i][j] = pre[i - 1][j - 1] ^ pre[i - 1][j] ^ pre[i][j - 1] ^ matrix[i - 1][j - 1]
4. i , j 的取值范围：[1, m] 和 [1, n]

代码

class Solution{
    public int KthLargesValue(int[][] matrix, int k){
        int m = matrix.length;
        int n = matrix[0].length;
        
        // 二维前缀和数组
        int[][] pre = new int[m + 1][n + 1];
        
        // 将所有的值都放入list中
        List<Integer> results = new ArrayList<>();
        for(int i = 1; i <= m; i ++){
            for(int j = 1; j <= n; j ++){
                pre[i][j] = pre[i-1][j-1] ^ pre[i-1][j] ^ pre[i][j-1] ^ matrix[i-1][j-1];
                results.add(pre[i][j]);
            }
        }
        
        // 排序，返回结果值
        Collections.sort(results);
        int size = results.size();
        return results.get(size - k);
    }
}