POJ3264 - Balanced Lineup （线段树 基本操作）

题目链接 : POJ3264 - Balanced Lineup

思路

这道题是一个线段树的基本应用，将树节点设置为如下：

struct Node
{
    int l, r; // 区间的起点和终点
    int ls, rs; // 左右孩子，所处的位置
    int maxn, minn; // 区间最大最小值
}

代码

#include 
#include 
#include 

using namespace std;

struct tNode
{
    int l, r;
    int ls, rs;
    int maxn, minn;
}tree[400100];
int nCount = 1;
int num[200100];

void build(int root, int l, int r)
{
    if(l==r)
    {
        // 区间长度为 1 时
        tree[root].maxn = tree[root].minn = num[l];
        tree[root].l = tree[root].r = l;
    }
    else
    {
        // 设置区间范围
        tree[root].l = l;
        tree[root].r = r;
        // 分配子节点
        int ls = tree[root].ls = ++nCount;
        int rs = tree[root].rs = ++nCount;
        // 递归build左右子树
        build(ls, l, (l+r)/2);
        build(rs, (l+r)/2+1, r);
        // 根据左右子树返回的结果，设置最大最小值
        tree[root].maxn = max(tree[ls].maxn, tree[rs].maxn);
        tree[root].minn = min(tree[ls].minn, tree[rs].minn);
    }
}


void query(int root, int l, int r, pair<int, int> & ans)
{
    if(tree[root].l==l&&tree[root].r==r)
    {
        // 所求范围与当前范围吻合
        ans.first = max(ans.first, tree[root].maxn);
        ans.second = min(ans.second, tree[root].minn);
    }
    else if(l>=tree[tree[root].rs].l)
    {
        // 全部位于右子树
        query(tree[root].rs, l, r, ans);
    }
    else if(r<=tree[tree[root].ls].r)
    {
        // 全部位于左子树
        query(tree[root].ls, l, r, ans);
    }
    else
    {
        // 横跨两个子树
        query(tree[root].ls, l, tree[tree[root].ls].r, ans);
        query(tree[root].rs, tree[tree[root].rs].l, r, ans);
    }
}

int main()
{
    int n, q;
    int l, s;
    pair<int, int> ans; // 可以直接用两个全局变量代替

    scanf("%d%d", &n, &q);
    for(int i=1; i<=n; i++) scanf("%d", num+i);

    build(1, 1, n);

    for(int i=1; i<=q; i++)
    {
        scanf("%d%d", &l, &s);
        ans = pair<int, int>(0, 0x7fffffff);
        query(1, l, s, ans);
        printf("%d\n", ans.first-ans.second);
    }
    return 0;
}