蓝桥杯-2014-B组-10-小朋友排队(拓展树状数组模板)

使用到了普通的树状数组和拓展的树状数组。
普通的只能单点修改和区间查询,利用两次区间查询可以做到单点查询。如果要区间修改时间复杂度是O(n)。
拓展树状数组可以区间修改和区间查询,利用两次区间操作即可实现单点操作,但增加了一个额外的数组。
题目思路是用一个普通bit记录该点是否未排序。一个拓展bit记录被排序次数。

  1. 然后模拟插入排序,选取最小点,该点左边的点被排序次数都+1(已排序的点可能会被再次累加,但不影响,原因见第4步)。
  2. 该点的被排序次数+n,n显然等于该点左边未排序的点的数量。
  3. 接着更新该点的未排序标志。
  4. 接着该点将不会参与后续排序,所以可以直接累加该点;由于后续不会再使用该点的被累加次数,所以再次执行第1步时,错误累加该点的被排序次数不会对结果造成影响。
#pragma warning(disable:4996)
#include <stdio.h>
#include <stdlib.h>
#include <utility>
#include <algorithm>

using namespace std;

typedef unsigned long long ull;

int isDebug = 0;

pair<int, int> childs[20]; // hight, index
int unhappy[20 + 1]; // 拓展bit 记录被排序次数
int unhappy2[20 + 1];
int counts[20 + 1]; // 普通bit 记录该点是否未排序

// 普通bit 单点修改 bit[i] += x
void bitAdd(int* bit, int i, int x)
{
    while (i < 21) {
        bit[i] += x;
        i += i & -i;
    }
}

// 拓展bit 区间修改 bit[i, N] += x
void bitAdd(int* bit1, int* bit2, int i, int x)
{
    int ii = i;
    while (i < 21) {
        bit1[i] += x;
        bit2[i] += x * (ii - 1);
        i += i & -i;
    }
}

// 普通bit 区间查询 前i项和 
int bitSum(int* bit, int i)
{
    int s = 0;
    while (i > 0) {
        s += bit[i];
        i -= i & -i;
    }
    return s;
}

// 拓展bit 区间查询 前i项和
int bitSum(int* bit1, int* bit2, int i)
{
    int s = 0;
    int ii = i;

    while (i > 0) {
        s += ii * bit1[i] - bit2[i];
        i -= i & -i;
    }

    return s;
}

// 遍历单点查询
void debug()
{
    if (!isDebug)
        return;

    int i, last = 0, now = 0;
    for (i = 1; i < 6; i++) {
        now = bitSum(unhappy, unhappy2, i);
        printf("%d\n", now - last);
        last = now;
    }
    printf("\n");
}

int main(void)
{
    freopen("q.txt", "r", stdin);

    int n, N, h, index;
    int last = 0, now = 0, ans = 0;

    scanf("%d", &N);

    for (n = 0; n < N; n++) {
        scanf("%d", &h);
        childs[n].first = h;
        childs[n].second = n + 1;
    }

    sort(childs, childs+n);

    for (n = 1; n < 21; n++)
        bitAdd(counts, n, 1);

    debug();
    for (n = 1; n < 21; n++)
        bitAdd(unhappy, unhappy2, n, 0);
    debug();

    for (n = 0; n < N; n++) {
        index = childs[n].second;

        // 1 o[1, index - 1] += 1
        bitAdd(unhappy, unhappy2, 1, 1);
        debug();
        bitAdd(unhappy, unhappy2, index, -1);
        debug();

        // 2 o[index] += index - n - 1
        int leftCount = bitSum(counts, index-1);
        bitAdd(unhappy, unhappy2, index, leftCount);
        debug();
        bitAdd(unhappy, unhappy2, index + 1, -leftCount);
        debug();

        // 3 计数数组中删除该节点
        bitAdd(counts, index, -1);
        // printf("%d %d\n", index, bitSum(unhappy, unhappy2, index) - bitSum(unhappy, unhappy2, index - 1));
        // 4
        index = bitSum(unhappy, unhappy2, index) - bitSum(unhappy, unhappy2, index - 1);
        ans += (1 + index) * index / 2;
    }

    printf("%d\n", ans);

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

推荐阅读更多精彩内容