poi 2182 2828 树状数组倍增查询

最新推荐文章于 2024-12-22 11:47:22 发布

衛宮

最新推荐文章于 2024-12-22 11:47:22 发布

阅读量553

点赞数 1

CC 4.0 BY-SA版权

分类专栏：数据结构技巧

本文链接：https://blog.youkuaiyun.com/u013482363/article/details/80739442

数据结构同时被 2 个专栏收录

21 篇文章

订阅专栏

技巧

2 篇文章

订阅专栏

本文介绍了一种查找特定前缀和位置的算法实现，通过树状数组结合二分搜索或倍增法，达到高效的查找目的。文章提供了具体的代码示例，并解释了其工作原理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这种题的主要目的是找位置, 比如第一个前缀和等于k的位置,

1. 可以用树状数组 + 二分, 时间为logn * logn, 这样时间稍微慢了些;

2. 用倍增法这片博文写得很好, http://www.hawstein.com/posts/binary-indexed-trees.html

int getKPos(int k){
        int bitMask = bthPow[upper_bound(bthPow, bthPow + MAXBTH, size) - bthPow - 1];
        int bitStart = 0;
        while(bitMask){
            if(bitStart + bitMask <= size){
                if(bit[bitStart + bitMask] < k){
                     k -= bit[bitStart + bitMask];
                     bitStart += bitMask;
                }
            }
            bitMask >>= 1;
        }
        return 1 + bitStart;// 注意这里的加一, 因为当前的前缀和是不满足等于k的最大位置
}