|
54 | 54 |
|
55 | 55 | <!-- 这里可写通用的实现逻辑 -->
|
56 | 56 |
|
| 57 | +二维前缀异或,然后求第 k 大的值即可。 |
| 58 | + |
57 | 59 | <!-- tabs:start -->
|
58 | 60 |
|
59 | 61 | ### **Python3**
|
60 | 62 |
|
61 | 63 | <!-- 这里可写当前语言的特殊实现逻辑 -->
|
62 | 64 |
|
63 | 65 | ```python
|
64 |
| - |
| 66 | +class Solution: |
| 67 | + def kthLargestValue(self, matrix: List[List[int]], k: int) -> int: |
| 68 | + m, n = len(matrix), len(matrix[0]) |
| 69 | + s = [[0] * (n + 1) for _ in range(m + 1)] |
| 70 | + ans = [] |
| 71 | + for i in range(m): |
| 72 | + for j in range(n): |
| 73 | + s[i + 1][j + 1] = s[i + 1][j] ^ s[i][j + 1] ^ s[i][j] ^ matrix[i][j] |
| 74 | + ans.append(s[i + 1][j + 1]) |
| 75 | + return heapq.nlargest(k, ans)[-1] |
65 | 76 | ```
|
66 | 77 |
|
67 | 78 | ### **Java**
|
68 | 79 |
|
69 | 80 | <!-- 这里可写当前语言的特殊实现逻辑 -->
|
70 | 81 |
|
71 | 82 | ```java
|
| 83 | +class Solution { |
| 84 | + |
| 85 | + public int kthLargestValue(int[][] matrix, int k) { |
| 86 | + int m = matrix.length, n = matrix[0].length; |
| 87 | + int[][] s = new int[m + 1][n + 1]; |
| 88 | + List<Integer> ans = new ArrayList<>(); |
| 89 | + for (int i = 0; i < m; ++i) { |
| 90 | + for (int j = 0; j < n; ++j) { |
| 91 | + s[i + 1][j + 1] = |
| 92 | + s[i + 1][j] ^ s[i][j + 1] ^ s[i][j] ^ matrix[i][j]; |
| 93 | + ans.add(s[i + 1][j + 1]); |
| 94 | + } |
| 95 | + } |
| 96 | + Collections.sort(ans); |
| 97 | + return ans.get(ans.size() - k); |
| 98 | + } |
| 99 | +} |
| 100 | + |
| 101 | +``` |
| 102 | + |
| 103 | +### **C++** |
| 104 | + |
| 105 | +```cpp |
| 106 | +class Solution { |
| 107 | +public: |
| 108 | + int kthLargestValue(vector<vector<int>>& matrix, int k) { |
| 109 | + int m = matrix.size(), n = matrix[0].size(); |
| 110 | + vector<vector<int>> s(m + 1, vector<int>(n + 1)); |
| 111 | + vector<int> ans; |
| 112 | + for (int i = 0; i < m; ++i) |
| 113 | + { |
| 114 | + for (int j = 0; j < n; ++j) |
| 115 | + { |
| 116 | + s[i + 1][j + 1] = s[i + 1][j] ^ s[i][j + 1] ^ s[i][j] ^ matrix[i][j]; |
| 117 | + ans.push_back(s[i + 1][j + 1]); |
| 118 | + } |
| 119 | + } |
| 120 | + sort(ans.begin(), ans.end()); |
| 121 | + return ans[ans.size() - k]; |
| 122 | + } |
| 123 | +}; |
| 124 | +``` |
72 | 125 |
|
| 126 | +### **Go** |
| 127 | +
|
| 128 | +```go |
| 129 | +func kthLargestValue(matrix [][]int, k int) int { |
| 130 | + m, n := len(matrix), len(matrix[0]) |
| 131 | + s := make([][]int, m+1) |
| 132 | + for i := range s { |
| 133 | + s[i] = make([]int, n+1) |
| 134 | + } |
| 135 | + var ans []int |
| 136 | + for i := 0; i < m; i++ { |
| 137 | + for j := 0; j < n; j++ { |
| 138 | + s[i+1][j+1] = s[i+1][j] ^ s[i][j+1] ^ s[i][j] ^ matrix[i][j] |
| 139 | + ans = append(ans, s[i+1][j+1]) |
| 140 | + } |
| 141 | + } |
| 142 | + sort.Ints(ans) |
| 143 | + return ans[len(ans)-k] |
| 144 | +} |
73 | 145 | ```
|
74 | 146 |
|
75 | 147 | ### **...**
|
|
0 commit comments