摘要:表現在二進制上的規律每次加上最末尾的求末尾的就拿這個數和它的補碼于。還有要求不是,要轉換一下,和之前那道的思路差不多。這里用一個的表示從到的和。
Range Sum Query - Immutable
Given an integer array nums, find the sum of the elements between indices i and j (i ≤ j), inclusive.
Example: Given nums = [-2, 0, 3, -5, 2, -1]
sumRange(0, 2) -> 1
sumRange(2, 5) -> -1
sumRange(0, 5) -3Note:
You may assume that the array does not change.
There are many calls to sumRange function.
用dp解。dp[i]表示sumRange(0, i-1)
function是dp[i+1] = dp[i] + nums[i]
所以sumRange(i, j) = dp[j+1] - dp[i]
public class NumArray { int[] dp; public NumArray(int[] nums) { dp = new int[nums.length + 1]; for(int i = 0; i < nums.length; i++) { dp[i+1] = dp[i] + nums[i]; } } public int sumRange(int i, int j) { return dp[j + 1] - dp[i]; } }Range Sum Query - Mutable
Given an integer array nums, find the sum of the elements between indices i and j (i ≤ j), inclusive.
The update(i, val) function modifies nums by updating the element at index i to val.
Example: Given nums = [1, 3, 5]sumRange(0, 2) -> 9
update(1, 2)
sumRange(0, 2) -> 8Note:
The array is only modifiable by the update function.
You may assume the number of calls to update and sumRange function is distributed evenly.
和上一題的區別在于,這次數組里面的值會動態的變化,所以不能和上面一題一樣,最開始初始化一個dp數組之后就不管了。如果還是用dp數組,那么每次update之后都得重新再改一遍,O(N)時間復雜度太高。
用binary index tree來做,update和sum的時間復雜度都是O(NlogN)。原理如下:
index: 1------2------3------4------5------6------7-----8-----9
binary: 0001-0010-0011-0100-0101-0110-0111-1000-1001
value: 2------5------6------3------1------7------5-----3-----2
tree[i]: [1,1]--[1,2]--[3,3]--[1,4]--[5,5]--[5,6]--[7,7]--[1,8]--[9,9]
Level1: 2------7------------16------------------------34
Level2: --------------6-------------1-----8-------------------9
right index in level1 has one "1" in binary representation.
right index in level2 has two "1" in binary representation.
......
2個主要的method:add 和 sum
add(idx, val): add val to nums[i]
sum(idx): get the sum from 1 to idx
add的時候改怎么操作呢?
舉個例子,比如add(5, 2), 我想在index = 5的地方加2,那么:
start從idx = 5開始,tree[5] += 2
接著,要更新tree[6],也就是sum(5, 6)
再接著更新tree[8],也就是sum(1, 8)
在圖里的表現就是我先從start開始,把start同一level所有往右相鄰的更新一遍,沒有相鄰的之后,就往上一面一層走,重復更新的過程。直到idx超過長度。
表現在二進制上的規律:
5 -------> 6 ------> 8
0101 -> 0110 -> 1000
每次加上最末尾的‘1’
0101 + 0001 = 0110
0110 + 0010 = 1000
求末尾的‘1’就拿這個數和它的補碼于。
假設這個數是x = n(1,0)-1-m(0)
m(0)表示這個數末尾有m個0,n(1,0)表示這個數前面有n個1或者0。那么第一個1就出現在從右數第m+1位上。
現在求x的反碼,~x = ~n(1,0)-0-m(1)
那么x的補碼就是:-x = ~n(1,0)-1-m(0)
所以: x&-x = n(0)-1-m(0)
void add(int idx, int val) { while(idx < tree.length) { tree[idx] += val; idx += i & -i; } }
int sum(int idx) { int result = 0; while(idx > 0) { result += tree[idx]; idx += i & -i; } return result; }
sum的方法同理,init的直接用add即可。這道題注意下數組的index是從0開始,所以每次要+1。還有要求update不是add,要轉換一下,update(idx, val) = add(idx, val - nums[idx])
public class NumArray { // binary index tree int[] tree; int[] nums; public NumArray(int[] nums) { this.nums = nums; tree = new int[nums.length + 1]; for(int i = 0; i < nums.length; i++) { add(i, nums[i]); } } private void add(int i, int val) { i += 1; while(i < tree.length) { tree[i] += val; i += i & -i; } } public void update(int i, int val) { int diff = val - nums[i]; nums[i] = val; add(i, diff); } public int sumRange(int i, int j) { return sum(j) - sum(i - 1); } // return the sumRange(0, i) private int sum(int i) { i+= 1; int result = 0; while(i > 0) { result += tree[i]; i -= i & -i; } return result; } }Range Sum Query 2D - Immutable
Given a 2D matrix matrix, find the sum of the elements inside the rectangle defined by its upper left corner (row1, col1) and lower right corner (row2, col2).
Range Sum Query 2D
The above rectangle (with the red border) is defined by (row1, col1) = (2, 1) and (row2, col2) = (4, 3), which contains sum = 8.
Example:
Given matrix = [
[3, 0, 1, 4, 2],
[5, 6, 3, 2, 1],
[1, 2, 0, 1, 5],
[4, 1, 0, 1, 7],
[1, 0, 3, 0, 5] ]sumRegion(2, 1, 4, 3) -> 8
sumRegion(1, 1, 2, 2) -> 11
sumRegion(1, 2, 2, 4) -> 12
Note:
You may assume that the matrix does not change.
There are many calls to sumRegion function. You may assume that row1 ≤ row2 and col1 ≤ col2.
和之前那道1d的思路差不多。這里用一個2d的prefix array:
dp[i + 1] [j + 1] 表示從 [0, 0] 到 [i, j] 的和。function是:
dp[i + 1] [j + 1] = dp[i + 1] [j] + dp[i] [j + 1] - dp[i] [j] + matrix[i] [j]
所以sum(row1, col1, row2, col2) = dp[row2 + 1] [col2 + 1] - dp[row1] [col2 + 1] - dp[row2 + 1] [col1] + dp[row1] [col1]
public class NumMatrix { // dp int[][] dp; public NumMatrix(int[][] matrix) { if(matrix == null || matrix.length == 0 || matrix[0] == null || matrix[0].length == 0) return; dp = new int[matrix.length + 1][matrix[0].length + 1]; for(int i = 0; i < matrix.length; i++) { for(int j = 0; j < matrix[0].length; j++) { dp[i+1][j+1] = dp[i+1][j] + dp[i][j+1] - dp[i][j] + matrix[i][j]; } } } public int sumRegion(int row1, int col1, int row2, int col2) { return dp[row2 + 1][col2 + 1] - dp[row1][col2 + 1] - dp[row2 + 1][col1] + dp[row1][col1]; } }Range Sum Query 2D - Mutable
Given a 2D matrix matrix, find the sum of the elements inside the rectangle defined by its upper left corner (row1, col1) and lower right corner (row2, col2).
Range Sum Query 2D
The above rectangle (with the red border) is defined by (row1, col1) = (2, 1) and (row2, col2) = (4, 3), which contains sum = 8.Example:
Given matrix = [
[3, 0, 1, 4, 2],
[5, 6, 3, 2, 1],
[1, 2, 0, 1, 5],
[4, 1, 0, 1, 7],
[1, 0, 3, 0, 5]
]sumRegion(2, 1, 4, 3) -> 8
update(3, 2, 2)
sumRegion(2, 1, 4, 3) -> 10Note:
The matrix is only modifiable by the update function.
You may assume the number of calls to update and sumRegion function is distributed evenly.
You may assume that row1 ≤ row2 and col1 ≤ col2.
和1d的mutable的差不多。還是用binary index tree來做。時間還是都是O(NlogN)。
文章版權歸作者所有,未經允許請勿轉載,若此文章存在違規行為,您可以聯系管理員刪除。
轉載請注明本文地址:http://specialneedsforspecialkids.com/yun/66563.html
摘要:假設有一個整數數組,計算下標從到包含和的數字的和。求和的請求將會在同一個整數數組上多次請求。這一題思路很簡單,因為。而利用動態規劃則很容易知道。這里將原先的一維數組替換成二維數組。要求計算一個矩形內的所有元素的值。 Range Sum Query Immutable Given an integer array nums, find the sum of the elements be...
摘要:題目要求可以先參考數組不發生變動時的題目。最后的葉節點為當前數組的值,非葉結點則記錄了子數組的范圍以及該子數組中所有元素的和。它是一個非常輕量級的數據結構,而且幾乎就是為這種題目量身打造。 題目要求 Given an integer array nums, find the sum of the elements between indices i and j (i ≤ j), inc...
Problem Given a 2D matrix matrix, find the sum of the elements inside the rectangle defined by its upper left corner (row1, col1) and lower right corner (row2, col2). https://leetcode.com/static/i...s...
Problem Given an integer array nums, find the sum of the elements between indices i and j (i ≤ j), inclusive. Example Given nums = [-2, 0, 3, -5, 2, -1] sumRange(0, 2) -> 1 sumRange(2, 5) -> -1 sumRan...
閱讀 3398·2023-04-25 22:04
閱讀 2197·2021-11-22 15:29
閱讀 2161·2021-10-11 10:57
閱讀 1402·2021-09-24 09:48
閱讀 3147·2021-09-09 09:34
閱讀 2543·2021-09-02 15:21
閱讀 2393·2019-08-30 15:53
閱讀 1120·2019-08-30 14:07