找出第 K 大的异或坐标值

找出第 K 大的异或坐标值

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

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

• 思路与算法

我们用 ⊕ 表示按位异或运算。

由于「按位异或运算」与「加法运算」有着十分相似的性质，它们都满足交换律：a⊕b=b⊕a 以及结合律：(a⊕b)⊕c=a⊕(b⊕c)

因此我们可以使用「前缀和」这一技巧对按位异或运算的结果进行维护。本题使用二维前缀和。

设二维前缀和 pre(i,j) 表示矩阵 matrix 中所有满足 0≤x<i 且 0≤y<j 的元素执行按位异或运算的结果。则有：pre(i,j)=pre(i−1,j)⊕pre(i,j−1)⊕pre(i−1,j−1)⊕matrix(i,j)

why?

以3*3矩阵为例，对于位于matrix(2,2)(9)处的数来说，其对于的pre(2,2)等于前8数依次做异或再与其异或，即5⊕2⊕3⊕1⊕6⊕2⊕7⊕4⊕9。对于matrix(2,1)(4)对应的pre(2,1)=5⊕2⊕1⊕6⊕7⊕4，matrix(1,2)(2)对应的pre(1,2)=5⊕2⊕3⊕1⊕6⊕2，matrix(1,1)(6)对应的pre(1,2)=5⊕2⊕1⊕6，有pre(2,1)⊕pre(1,2) = 5⊕2⊕1⊕6⊕7⊕4⊕5⊕2⊕3⊕1⊕6⊕2=7⊕4⊕3⊕2(因为对一个数 x 异或两次 y，结果仍然为 x 本身，即：x⊕y⊕y=x），所以pre(2,2) = pre(1,2)⊕pre(2,1)⊕pre(1,1)⊕matrix(2,2)

代码实现：

class Solution {
public:
    int kthLargestValue(vector<vector<int>>& matrix, int k) {
        int m = matrix.size(), n = matrix[0].size();
        vector<vector<int>> pre(m + 1, vector<int>(n + 1));
        vector<int> results;
        for (int i = 1; i <= m; ++i) {
            for (int j = 1; j <= n; ++j) {
                pre[i][j] = pre[i - 1][j] ^ pre[i][j - 1] ^ pre[i - 1][j - 1] ^ matrix[i - 1][j - 1];
                results.push_back(pre[i][j]);
            }
        }

        sort(results.begin(), results.end(), greater<int>());
        return results[k - 1];
    }
};