树状数组理解

数状数组理解:

1,为了解决什么问题

频繁查询数组前缀和+数组元素动态更新时

2,原理

本质上利用数字的二进制思想来存储前缀和

举个例子:

原数组arry[1...n]

数状数组bit_tree[1...n]

bit_tree[1] = arry[1]

bit_tree[2] = arry[1] + arry[2]

....

bit_tree[i]本质上存储sum = arry[i - lowbit(i) + 1] +... arry[i]

3,这样的好处是什么呢?

正常数组比如arry[1] += 1时,后面所有的idx都得做+1操作;用数组数组只需要更新几个节点即可,降低了时间复杂度~

来做题消化一下~

https://leetcode.cn/problems/minimum-inversion-count-in-subarrays-of-fixed-length

思路:

1,滑动窗口来计算当前当前逆序值~

2,使用树状数组来降低时间复杂度~

objectivec 复制代码
typedef struct {
    int* tree;
    int size;
} FenwickTree;

FenwickTree* createFenwickTree(int size) {
    FenwickTree* ft = (FenwickTree*)malloc(sizeof(FenwickTree));
    ft->size = size + 2; 
    ft->tree = (int*)calloc(ft->size, sizeof(int));
    return ft;
}

void update(FenwickTree* ft, int idx, int delta) {
    idx++; 
    while (idx < ft->size) {
        ft->tree[idx] += delta;
        idx += idx & -idx;
    }
}

int query(FenwickTree* ft, int idx) {
    idx++; 
    int sum = 0;
    while (idx > 0) {
        sum += ft->tree[idx];
        idx -= idx & -idx;
    }
    return sum;
}

int cmp(const void* a, const void* b) 
{ return *(int*)a - *(int*)b; }

void process_data_qsort_Deduplicate(int* nums, int n, int* processed_index, int* qsorted_data) {
    int* sorted = (int*)malloc(n * sizeof(int));
    memcpy(sorted, nums, n * sizeof(int));
    
    qsort(sorted, n, sizeof(int), cmp);
    
    int m = 0;
    for (int i = 0; i < n; i++) {
        if (i == 0 || sorted[i] != sorted[i - 1]) {
            qsorted_data[m++] = sorted[i];
        }
    }
    
    for (int i = 0; i < n; i++) {
        int left = 0, right = m - 1;
        while (left <= right) {
            int mid = left + (right - left) / 2;
            if (qsorted_data[mid] == nums[i]) {
                processed_index[i] = mid;
                break;
            } else if (qsorted_data[mid] < nums[i]) {
                left = mid + 1;
            } else {
                right = mid - 1;
            }
        }
    }
    
    free(sorted);
}
long long minInversionCount(int* nums, int numsSize, int k) {
    if (k == 1) return 0; // 单个元素没有逆序对
    int  n = numsSize;
    int* processed_index = (int*)malloc(n * sizeof(int));
    int* qsorted_data = (int*)malloc(n * sizeof(int));
    process_data_qsort_Deduplicate(nums, n, processed_index, qsorted_data);
    int* timberavos = (int*)malloc(n * sizeof(int));
    memcpy(timberavos, processed_index, n * sizeof(int));
    FenwickTree* ft = createFenwickTree(n);
    long long current_Inversion = 0;
    for (int i = 0; i < k; i++) {
        int greater_count = query(ft, n - 1) - query(ft, timberavos[i]);
        current_Inversion += greater_count;
        update(ft, timberavos[i], 1);
    }
    long long min_Inversion = current_Inversion;
    for (int i = k; i < n; i++) {
        int left_val = timberavos[i - k];
        update(ft, timberavos[i-k], -1);
        int left_greater = query(ft, timberavos[i-k]-1);
        current_Inversion -= left_greater;
        // printf("current_Inversion00:%d %d\n", i, current_Inversion);
        int right_val = timberavos[i];
        update(ft, timberavos[i], 1);
        int right_greater = query(ft, n - 1) - query(ft, timberavos[i]);
        current_Inversion += right_greater;
        // printf("current_Inversion11:%d %d\n", i, current_Inversion);
        if (current_Inversion < min_Inversion) {
            min_Inversion = current_Inversion;
        }
    }
    
    free(processed_index);
    free(qsorted_data);
    free(timberavos);
    free(ft->tree);
    free(ft);
    
    return min_Inversion;
}
相关推荐
爱写代码的小朋友7 分钟前
基于多约束遗传算法的中小学排座位优化模型研究
linux·人工智能·算法
один but you9 分钟前
unorder_map 和unorder_set
算法·哈希算法·散列表
sheeta199822 分钟前
LeetCode 每日一题笔记 日期:2026.05.20 题目:2657. 找到前缀公共数组
笔记·算法·leetcode
数智工坊39 分钟前
【UniT论文阅读】:用统一物理语言打通人类与人形机器人的知识壁垒
论文阅读·人工智能·深度学习·算法·机器人
梓䈑41 分钟前
【算法题攻略】模拟
c++·算法
Evand J43 分钟前
【课题推荐与代码介绍】卡尔曼滤波器正反向估计算法原理与MATLAB实现
开发语言·算法·matlab
DFT计算杂谈1 小时前
VASP新手入门: IVDW 色散修正参数
linux·运维·服务器·python·算法
吃着火锅x唱着歌1 小时前
LeetCode 962.最大宽度坡
算法·leetcode·职场和发展
无限进步_1 小时前
【C++】C++11的类功能增强与STL变化
java·前端·数据结构·c++·后端·算法
WL_Aurora1 小时前
Python 算法基础篇之排序算法(一):冒泡、选择、插入
python·算法·排序算法