leetcode 1738.找出第K大的异或坐标值 - 前缀和

leetcode 1738.找出第K大的异或坐标值 - 前缀和

题干

给你一个二维矩阵 matrix 和一个整数 k ，矩阵大小为 m x n 由非负整数组成。
矩阵中坐标 (a, b) 的 值 可由对所有满足 0 <= i <= a < m 且 0 <= j <= b < n 的元素 matrix[i][j]（下标从 0 开始计数）执行异或运算得到。
请你找出 matrix 的所有坐标中第 k 大的值（k 的值从 1 开始计数）。

示例 1：
输入：matrix = [[5,2],[1,6]], k = 1
输出：7
解释：坐标 (0,1) 的值是 5 XOR 2 = 7 ，为最大的值。

示例 2：
输入：matrix = [[5,2],[1,6]], k = 2
输出：5
解释：坐标 (0,0) 的值是 5 = 5 ，为第 2 大的值。

示例 3：
输入：matrix = [[5,2],[1,6]], k = 3
输出：4
解释：坐标 (1,0) 的值是 5 XOR 1 = 4 ，为第 3 大的值。

示例 4：
输入：matrix = [[5,2],[1,6]], k = 4
输出：0
解释：坐标 (1,1) 的值是 5 XOR 2 XOR 1 XOR 6 = 0 ，为第 4 大的值。

提示：
m == matrix.length
n == matrix[i].length
1 <= m, n <= 1000
0 <= matrix[i][j] <= 106
1 <= k <= m * n

知识点&算法

读完题可以发现，所谓坐标的值，就是指以矩阵左上角为左上角，坐标为右下角的矩阵内部全部元素的异或结果。
显然可以用前缀和的思想处理，只是要考虑两维。
然后通过优先队列来获得第k大的元素。

题解

前缀异或+优先队列:

class Solution {
public:
    int kthLargestValue(vector<vector<int>>& matrix, int k) {
        priority_queue<int,vector<int>,less<int>> vals;
        int n = matrix.size() , m = matrix[0].size();
        int val[n+5][m+5];
        for(int i = 0 ; i < n ; ++i){
            for(int j = 0 ; j < m ; ++j){
                val[i][j] = matrix[i][j];
                if(j > 0) val[i][j] ^= val[i][j-1];
                if(i > 0) val[i][j] ^= val[i-1][j];
                if(i > 0 && j > 0) val[i][j] ^= val[i-1][j-1];
                vals.push(val[i][j]);
            }
        }
        while(--k){
            vals.pop();
        }
        return vals.top();
    }
};