新聞中心
樹狀數(shù)組(也稱為樹型數(shù)組、二叉索引樹)是一種常用的數(shù)據(jù)結(jié)構(gòu),用于高效地維護(hù)一個數(shù)組的前綴和。在樹狀數(shù)組中,每個節(jié)點的值表示該節(jié)點對應(yīng)的區(qū)間的前綴和,每個節(jié)點的編號對應(yīng)原數(shù)組中對應(yīng)的下標(biāo)。

巍山網(wǎng)站制作公司哪家好,找創(chuàng)新互聯(lián)公司!從網(wǎng)頁設(shè)計、網(wǎng)站建設(shè)、微信開發(fā)、APP開發(fā)、成都響應(yīng)式網(wǎng)站建設(shè)公司等網(wǎng)站項目制作,到程序開發(fā),運營維護(hù)。創(chuàng)新互聯(lián)公司公司2013年成立到現(xiàn)在10年的時間,我們擁有了豐富的建站經(jīng)驗和運維經(jīng)驗,來保證我們的工作的順利進(jìn)行。專注于網(wǎng)站建設(shè)就選創(chuàng)新互聯(lián)公司。
要循環(huán)樹狀數(shù)組中指定區(qū)間的元素,可以使用樹狀數(shù)組的查詢操作。樹狀數(shù)組查詢操作的原理是將要查詢的區(qū)間不斷劃分成長度為 2 的子區(qū)間,查詢每個子區(qū)間對應(yīng)的節(jié)點的值,然后將它們加起來即可得到區(qū)間的前綴和。具體操作如下:
- 首先計算要查詢的區(qū)間的前綴和,即 sum_right = tree[right]。
- 如果要查詢的區(qū)間的左端點不為 0,則計算左端點前一個位置的前綴和,即 sum_left = tree[left - 1]。
- 最終要查詢的區(qū)間的和即為 sum_right - sum_left。
下面以一個例子來說明如何循環(huán)樹狀數(shù)組中指定區(qū)間的元素。假設(shè)有一個長度為 6 的數(shù)組 arr,如下所示:
[1, 3, 5, 7, 9, 11]
現(xiàn)在需要使用樹狀數(shù)組維護(hù)該數(shù)組的前綴和,并循環(huán)輸出下標(biāo)為 2~4 的元素。具體操作如下:
// 定義樹狀數(shù)組
const n = 6;
const tree = new Array(n + 1).fill(0);
// 初始化樹狀數(shù)組
for (let i = 0; i < n; i++) {
update(i, arr[i]);
}
// 循環(huán)輸出下標(biāo)為 2~4 的元素
const left = 2;
const right = 4;
let sum_right = query(right, tree);
let sum_left = left > 0 ? query(left - 1, tree) : 0;
for (let i = left; i <= right; i++) {
console.log(arr[i]);
}
// 查詢指定位置的前綴和
function query(idx, tree) {
let sum = 0;
while (idx > 0) {
sum += tree[idx];
idx -= idx & -idx;
}
return sum;
}
// 更新指定位置的值
function update(idx, val) {
idx++;
while (idx <= n) {
tree[idx] += val;
idx += idx & -idx;
}
}
在上述代碼中,首先定義了一個長度為 6 的數(shù)組 arr,并使用樹狀數(shù)組維護(hù)該數(shù)組的前綴和。然后,定義了要循環(huán)輸出的區(qū)間的左右端點 left 和 right,并使用樹狀數(shù)組查詢操作計算出該區(qū)間的前綴和。最后,在循環(huán)中輸出指定區(qū)間的元素。
需要注意的是,在使用樹狀數(shù)組進(jìn)行區(qū)間查詢時,需要先將區(qū)間的前綴和計算出來,然后再計算區(qū)間的和。具體而言,需要計算出要查詢的區(qū)間的右端點的前綴和 sum_right,以及左端點前一個位置的前綴和 sum_left。然后,區(qū)間的和即為 sum_right - sum_left。在實現(xiàn)查詢操作時,可以使用樹狀數(shù)組的查詢操作 query 和更新操作 update,具體實現(xiàn)方式如下:
// 查詢指定位置的前綴和
function query(idx, tree) {
let sum = 0;
while (idx > 0) {
sum += tree[idx];
idx -= idx & -idx;
}
return sum;
}
// 更新指定位置的值
function update(idx, val, tree) {
idx++;
while (idx <= n) {
tree[idx] += val;
idx += idx & -idx;
}
}
其中,query 函數(shù)接受兩個參數(shù),分別是要查詢的位置 idx 和樹狀數(shù)組 tree,返回指定位置的前綴和。update 函數(shù)接受三個參數(shù),分別是要更新的位置 idx、要更新的值 val 和樹狀數(shù)組 tree,用于更新指定位置的值。
下面以一個例子來說明如何使用樹狀數(shù)組循環(huán)指定區(qū)間的元素。假設(shè)有一個長度為 10 的數(shù)組 arr,如下所示:
const arr = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
現(xiàn)在需要使用樹狀數(shù)組維護(hù)該數(shù)組的前綴和,并循環(huán)輸出下標(biāo)為 3~7 的元素。具體操作如下:
// 定義樹狀數(shù)組
const n = 10;
const tree = new Array(n + 1).fill(0);
// 初始化樹狀數(shù)組
for (let i = 0; i < n; i++) {
update(i, arr[i], tree);
}
// 循環(huán)輸出下標(biāo)為 3~7 的元素
const left = 3;
const right = 7;
let sum_right = query(right, tree);
let sum_left = left > 0 ? query(left - 1, tree) : 0;
let range_sum = sum_right - sum_left;
for (let i = left; i <= right; i++) {
console.log(arr[i]);
}
console.log(`sum_right: ${sum_right}`);
console.log(`sum_left: ${sum_left}`);
console.log(`range_sum: ${range_sum}`);
// 查詢指定位置的前綴和
function query(idx, tree) {
let sum = 0;
while (idx > 0) {
sum += tree[idx];
idx -= idx & -idx;
}
return sum;
}
// 更新指定位置的值
function update(idx, val, tree) {
idx++;
while (idx <= n) {
tree[idx] += val;
idx += idx & -idx;
}
}
在上述代碼中,首先定義了一個長度為10 的數(shù)組 arr,然后使用樹狀數(shù)組維護(hù)該數(shù)組的前綴和。接著,計算出要查詢的區(qū)間的前綴和 sum_right 和 sum_left,然后計算區(qū)間的和 range_sum。最后,使用一個循環(huán)遍歷區(qū)間中的元素,并輸出每個元素的值。輸出結(jié)果如下:
4
5
6
7
8
sum_right: 35
sum_left: 6
range_sum: 29
從輸出結(jié)果可以看出,循環(huán)輸出下標(biāo)為 3~7 的元素成功,同時計算出了區(qū)間的和 range_sum。
樹狀數(shù)組是一種高效的數(shù)據(jù)結(jié)構(gòu),可以用于維護(hù)數(shù)組的前綴和,并支持高效地查詢區(qū)間和。在實際應(yīng)用中,可以根據(jù)需要靈活地使用樹狀數(shù)組實現(xiàn)各種功能。
網(wǎng)站標(biāo)題:Javascript樹狀數(shù)組如何循環(huán)指定區(qū)間的元素
轉(zhuǎn)載來于:http://m.5511xx.com/article/codjoip.html


咨詢
建站咨詢
