Fenwick Tree/B.I.T树状数组算法

树状数组适用范围:给定区间,求最值,求和,区间单点修改。
与RMQ不同的是,RMQ一般只用作区间求最值。但在最值方面RMQ更加便捷。

树状数组

从图上可以看出要找到c[i]的父节点只要用i+x(x为i转化为二进制从右起第一个出现1的位置)就可以了
寻找x可以通过lowbit函数,即通过计算机补码的巧用,x=i&(-i)

int lowbit(int x)
{
    return x&(-x);
}

而每一个父节点存储的数据都是下面所有子节点的全部信息。对子节点修改就必定会对其所有的父节点进行更新,父节点c[i]下标的查找如下:

void update(int i, int x)
{
    while(i<=n)
    {
        c[i]+=x;
        i+=lowbit(i);//更新父节点
    }
}

对于查找1~i的和、最大值、最小值可以表示为:

int sum(int i)
{
    int res=0;
    while(i>0)
    {
        res+=c[i];
        i-=lowbit(i);
    }
    return res;
}

典型例题(模版):
HDU - 1166
对一个区域的数列进行查找、修改

#include<cstdio>
#include<cstring>
using namespace std;

int c[50005];
int n;

int lowbit(int x)
{
    return x&(-x);
}

void update(int i, int x)
{
    while(i<=n)
    {
        c[i]+=x;
        i+=lowbit(i);//更新父节点
    }
}

int sum(int i)
{
    int res=0;
    while(i>0)
    {
        res+=c[i];
        i-=lowbit(i);
    }
    return res;
}

int main()
{
    int T, cnt;
    scanf("%d", &T);
    for(cnt=1; cnt<=T; cnt++)
    {
        int i, temp;
        char op[10];
        memset(c,0,sizeof(c));

        scanf("%d", &n);
        for(i=1 ;i<=n; i++)
        {
            scanf("%d", &temp);
            update(i, temp);
        }
        printf("Case %d:\n", cnt);
        while(~scanf("%s", op))
        {
            int l, r, k;
            if(!strcmp(op,"End")) break;
            if(!strcmp(op,"Query"))
            {
                scanf("%d%d", &l, &r);
                printf("%d\n", sum(r)-sum(l-1));
            }
            if(!strcmp(op,"Add"))
            {
                scanf("%d%d", &k, &temp);
                update(k, temp);
            }
            if(!strcmp(op,"Sub"))
            {
                scanf("%d%d", &k, &temp);
                update(k, -temp);
            }
        }
    }
    return 0;
}

例题:POJ - 3928
题目大意:在数列X中找到a、b、c
满足a<b<c和X[a]<X[b]<X[c]
思路:求取前缀和和后缀和的问题,update(X[i], 1);再getsum()即可

#include<cstdio>
#include<cstring>
using namespace std;
#define ll long long int
const int maxn = 1e5+7;
const int max_num = 1e5;

int a[maxn];
ll c[maxn];

int pos_west[maxn], pos_east[maxn], neg_west[maxn], neg_east[maxn];

int lowbit(int x)
{
    return x&(-x);
}

void update(int i, int p)
{
    while(i<=max_num)
    {
        c[i]+=p;
        i+=lowbit(i);
    }
}

ll getsum(int i)
{
    int res=0;
    while(i>0)
    {
        res+=c[i];
        i-=lowbit(i);
    }
    return res;
}

void init()
{
    memset(a,0,sizeof(a));
    memset(c,0,sizeof(c));
    memset(pos_west,0,sizeof(pos_west));
    memset(pos_east,0,sizeof(pos_east));
    memset(neg_west,0,sizeof(neg_west));
    memset(neg_east,0,sizeof(neg_east));
}

int main()
{
    int T;
    scanf("%d", &T);
    while(T--)
    {
        int i, n;
        ll sum=0;
        scanf("%d", &n);
        init();

        for(i=1; i<=n; i++)
        {
            scanf("%d", &a[i]);
            update(a[i],1);
            pos_west[i]=getsum(a[i]-1);
            pos_east[i]=getsum(max_num)-getsum(a[i]);
        }

        memset(c,0,sizeof(c));
        for(i=n; i>0; i--)
        {
            update(a[i],1);
            neg_west[i]=getsum(a[i]-1);
            neg_east[i]=getsum(max_num)-getsum(a[i]);
        }

        for(i=1; i<=n; i++)
        {
            sum+=pos_west[i]*neg_east[i]+neg_west[i]*pos_east[i];
        }
        printf("%I64d\n", sum);
    }
    return 0;
}

注:此方法还可以用来求取逆序数!!!

例题:HDU - 1556
给定区间,每次在这个区间加1,求每个点一共加了多少次1
思路:前缀和思想,如图

前缀和
#include<cstdio>
#include<cstring>
using namespace std;

int c[100005];
int n;

int lowbit(int x)
{
    return x&(-x);
}

void update(int i, int p)
{
    while(i<=n)
    {
        c[i]+=p;
        i+=lowbit(i);
    }
}

int getsum(int i)
{
    int res=0;
    while(i>0)
    {
        res+=c[i];
        i-=lowbit(i);
    }
    return res;
}

int main()
{
    while(~scanf("%d", &n)&&n)
    {
        int a, b;
        memset(c,0,sizeof(c));
        for(int i=1; i<=n; i++)
        {
            scanf("%d%d", &a, &b);
            update(a, 1);
            update(b+1, -1);
        }
        for(int i=1; i<=n; i++)
        {
            if(i!=1) printf(" ");
            printf("%d", getsum(i));
        }
        printf("\n");
    }
    return 0;
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,128评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,316评论 3 388
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,737评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,283评论 1 287
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,384评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,458评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,467评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,251评论 0 269
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,688评论 1 306
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,980评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,155评论 1 342
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,818评论 4 337
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,492评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,142评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,382评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,020评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,044评论 2 352

推荐阅读更多精彩内容

  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,644评论 18 139
  • 一、 C/C++程序基础 面试例题1——分析代码写输出(一般赋值语句的概念和方法)。 面试例题2—...
    LuckTime阅读 1,972评论 2 42
  • 归去来兮。 1.1 说明 本篇为《挑战程序设计竞赛(第2版)》[http://www.ituring.com.cn...
    尤汐Yogy阅读 14,318评论 0 160
  • 汉飞阅读 183评论 4 0
  • 标题 标题前面加上#号, 即代表是标题,1-6级标题,分别对应1-6个#号 列表 无序列表 在列表前面加上-,即可...
    cpf阅读 335评论 0 1