请你实现一个类 SubrectangleQueries ,它的构造函数的参数是一个 rows x cols 的矩形(这里用整数矩阵表示),并支持以下两种操作:
updateSubrectangle(int row1, int col1, int row2, int col2, int newValue)
用 newValue 更新以 (row1,col1) 为左上角且以 (row2,col2) 为右下角的子矩形。
getValue(int row, int col)
返回矩形中坐标 (row,col) 的当前值。
示例 1:
输入:
[“SubrectangleQueries”,“getValue”,“updateSubrectangle”,“getValue”,“getValue”,“updateSubrectangle”,“getValue”,“getValue”]
[[[[1,2,1],[4,3,4],[3,2,1],[1,1,1]]],[0,2],[0,0,3,2,5],[0,2],[3,1],[3,0,3,2,10],[3,1],[0,2]]
输出:
[null,1,null,5,5,null,10,5]
解释:
SubrectangleQueries subrectangleQueries = new SubrectangleQueries([[1,2,1],[4,3,4],[3,2,1],[1,1,1]]);
// 初始的 (4x3) 矩形如下:
// 1 2 1
// 4 3 4
// 3 2 1
// 1 1 1
subrectangleQueries.getValue(0, 2); // 返回 1
subrectangleQueries.updateSubrectangle(0, 0, 3, 2, 5);
// 此次更新后矩形变为:
// 5 5 5
// 5 5 5
// 5 5 5
// 5 5 5
subrectangleQueries.getValue(0, 2); // 返回 5
subrectangleQueries.getValue(3, 1); // 返回 5
subrectangleQueries.updateSubrectangle(3, 0, 3, 2, 10);
// 此次更新后矩形变为:
// 5 5 5
// 5 5 5
// 5 5 5
// 10 10 10
subrectangleQueries.getValue(3, 1); // 返回 10
subrectangleQueries.getValue(0, 2); // 返回 5
最多有 500 次updateSubrectangle 和 getValue 操作。
1 <= rows, cols <= 100
rows == rectangle.length
cols == rectangle[i].length
0 <= row1 <= row2 < rows
0 <= col1 <= col2 < cols
1 <= newValue, rectangle[i][j] <= 10^9
0 <= row < rows
0 <= col < cols
updateSubrectangle方法把操作存起来,getValue方法从后向前遍历所有操作,看是否操作到了这个值,如果操作到了,返回这个数变成了多少:
class SubrectangleQueries {
public:
SubrectangleQueries(vector<vector<int>>& rectangle) : rectangle(rectangle) { }
void updateSubrectangle(int row1, int col1, int row2, int col2, int newValue) {
op.push_back({row1, row2, col1, col2, newValue});
}
int getValue(int row, int col) {
int ans = 0;
vector<vector<int>>::reverse_iterator re = op.rend();
for (vector<vector<int>>::reverse_iterator rb = op.rbegin(); rb != re; ++rb) {
if (row >= (*rb)[0] && row <= (*rb)[1] && col >= (*rb)[2] && col <= (*rb)[3]) {
return (*rb)[4];
}
}
return rectangle[row][col];
}
private:
vector<vector<int>> rectangle;
vector<vector<int>> op;
};
/**
* Your SubrectangleQueries object will be instantiated and called as such:
* SubrectangleQueries* obj = new SubrectangleQueries(rectangle);
* obj->updateSubrectangle(row1,col1,row2,col2,newValue);
* int param_2 = obj->getValue(row,col);
*/
如果操作次数为n,getValue方法时间复杂度为O(n),空间复杂度为O(1),updateSubrectangle方法时间复杂度为O(1),空间复杂度为O(1)。