HDU 1754

最新推荐文章于 2019-10-25 00:39:07 发布

X-Wyatt

最新推荐文章于 2019-10-25 00:39:07 发布

阅读量379

点赞数

分类专栏： ACM Segment tree 文章标签： cmd CMD insert query Query struct Struct tree

本文链接：https://blog.csdn.net/X_White/article/details/7975074

版权

ACM 同时被 2 个专栏收录

103 篇文章 0 订阅

订阅专栏

Segment tree

3 篇文章 0 订阅

订阅专栏

线段树的模板题。

buildTree, insert, update, query 四个函数写的时候注意叶子节点，左右孩子就行了

#include <stdio.h>
#include <string.h>

#define LEN 400400
#define NMAX -999999

typedef struct Node {
    int l, r;
    struct Node* lChild;
    struct Node* rChild;
    int nMax, id;
}Node;
Node tree[LEN];
int score[LEN];
int nCount;
int nMax;

void buildTree(Node* pRoot, int l, int r) {
    pRoot->l = l;
    pRoot->r = r;
    pRoot->nMax = NMAX;
    if(l != r) {
        nCount++;
        pRoot->lChild = tree + nCount;
        nCount++;
        pRoot->rChild = tree + nCount;
        buildTree(pRoot->lChild, l, (l+r)/2);
        buildTree(pRoot->rChild, (l+r)/2+1, r);
    }
}

void insert(Node* pRoot, int i, int num) {
    if(pRoot->l == i && pRoot->r == i) {
        pRoot->nMax = num;
        pRoot->id = i;
        return;
    }
    if(pRoot->nMax < num) {
        pRoot->nMax = num;
        pRoot->id = i;
    }
    int mid = (pRoot->l + pRoot->r)/2;
    if(i <= mid) {
        insert(pRoot->lChild, i, num);
    }
    else {
        insert(pRoot->rChild, i, num);
    }
}

void query(Node* pRoot, int s, int e) {
    if(pRoot->l == s && pRoot->r == e) {
        if(nMax < pRoot->nMax) {
            nMax = pRoot->nMax;
        }
        return;
    }
    int mid = (pRoot->l + pRoot->r)/2;
    if(e <= mid) {
        query(pRoot->lChild, s, e);
    }
    else if(s >= mid+1) {
        query(pRoot->rChild, s, e);
    }
    else {
        query(pRoot->lChild, s, mid);
        query(pRoot->rChild, mid+1, e);
    }
}

void update(Node* pRoot, int id, int num) {
    int i;
    if(pRoot->l == id && pRoot->r == id) {
        pRoot->nMax = num;
        return;
    }
    if(pRoot->id == id) {
        int tMax = NMAX;
        int index;
        for(i = pRoot->l; i <= pRoot->r; i++) {
            if(tMax < score[i]) {
                tMax = score[i];
                index = i;
            }
        }
        pRoot->nMax = tMax;
        pRoot->id = index;
        if(num > pRoot->nMax) {
            pRoot->nMax = num;
            pRoot->id = id;
        }
    }
    else if(num > pRoot->nMax) {
        pRoot->nMax = num;
        pRoot->id = id;
    }
    int mid = (pRoot->l + pRoot->r)/2;
    if(id <= mid) {
        update(pRoot->lChild, id, num);
    }
    else if(id >= mid+1) {
        update(pRoot->rChild, id, num);
    }
}

int main() {
    int n, m;
    while(scanf("%d%d", &n, &m) != EOF) {
        int i, j, num;
        char cmd[10];
        nCount = 0;
        memset(tree, 0, sizeof(tree));
        memset(score, 0, sizeof(score));
        buildTree(tree, 1, n);
        for(i = 1; i <= n; i++) {
            scanf("%d", &num);
            score[i] = num;
            insert(tree, i, num);
        }
        for(i = 1; i <= m; i++) {
            int a, b;
            memset(cmd, 0, sizeof(cmd));
            scanf("%s%d%d", cmd, &a, &b);
            if('Q' == cmd[0]) {
                nMax = NMAX;
                query(tree, a, b);
                printf("%d\n", nMax);
            }
            else {
                score[a] = b;
                update(tree, a, b);
            }
        }
    }
    return 0;
}