POJ 4047 Garden(线段树)

参考题解
有n个连续的一维上的点,每个点有一个美丽值。
每次有三种操作:
1、将x点的美丽值改为y。
2、将x、y交换
3、查询[x y]间,连续k个点的美丽值总和的最大值。
这里我是看的题解的做法,利用线段树,叶子i表示[i i+k]美丽值总和,然后每个点维护一个区间最大值。每个位置向前取k个数,作为这个位置的结果,线段树维护下每个区间的最大值即可,当改变一个位置p 的数时,把他所影响的区间(p,p+k-1)更新一下即可,查询(x,y)区间时,查询(x+k-1,y)的最大值即可。

#include <iostream>
#include <algorithm>
#include <cstring>
#include <string>
#include <cstdio>
#define MAXN 222222
#define MAXM 22222
#define INF 1000000007
#define lch(x) x << 1
#define rch(x) x << 1 | 1
#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1
using namespace std;
int val[MAXN], t[MAXN], mx[4 * MAXN], cover[4 * MAXN];
int n, k, m;

//上推状态
void up(int rt)
{
    mx[rt] = max(mx[lch(rt)], mx[rch(rt)]);
}

void down(int rt)
{
    //如果是没有记录更改值,那么直接跳过
    if(!cover[rt])  return ;
    //否则当前节点的把更改标记推给左右孩子
    cover[lch(rt)] += cover[rt];
    cover[rch(rt)] += cover[rt];
    //孩子的值进行更改
    mx[lch(rt)] += cover[rt];
    mx[rch(rt)] += cover[rt];
    //去除自身标记
    cover[rt] = 0;
}

//建树
void build(int l, int r, int rt)
{
    //先将自身的标记置为0
    cover[rt] = 0;
    if(l == r)
    {
        mx[rt] = t[l];
        return;
    }
    int m = (l + r) >> 1;
    build(lson);
    build(rson);
    up(rt);
}

void update(int L, int R, int l, int r, int rt, int v)  //区间更新,v是增量
{
    if(L <= l && R >= r)
    {
        mx[rt] += v;
        cover[rt] += v;
        return;
    }
    down(rt);
    int m = (l + r) >> 1;
    if(m >= L) update(L, R, lson, v);
    if(m < R) update(L, R, rson, v);
    up(rt);
}

int query(int L, int R, int l, int r, int rt)
{
    if(L <= l && R >= r) return mx[rt];
    down(rt);
    int tmp = -INF;
    int m = (l + r) >> 1;
    if(m >= L) tmp = max(tmp, query(L, R, lson));
    if(m < R) tmp = max(tmp, query(L, R, rson));
    return tmp;
}

void change(int x, int y)
{
    //因为是每个叶节点都是代表了长度为k的和,所以一个点的值进行改变之后,一连k个值都要改
    int st = x - k + 1, ed = x;
    if(st < 1) st = 1;
    update(st, ed, 1, n, 1, y - val[x]);
    val[x] = y;
}

int main()
{
    int T;
    cin >> T;
    while(T--)
    {
        scanf("%d%d%d", &n, &m, &k);
        for(int i = 1; i <= n; i++) scanf("%d", &val[i]);
        t[1] = 0;
        for(int i = 1; i <= k; i++) t[1] += val[i];
        for(int i = 2; i <= n - k + 1; i++) t[i] = t[i - 1] - val[i - 1] + val[i - 1 + k];
        n = n - k + 1;
        build(1, n, 1);
        int op, x, y;
        while(m--)
        {
            scanf("%d%d%d", &op, &x, &y);
            if(op == 0) change(x, y);
            else if(op == 1)
            {
                int tmp = val[x];
                change(x, val[y]);
                change(y, tmp);
            }
            else if(op == 2) printf("%d\n", query(x, y - k + 1, 1, n, 1));
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_40788897/article/details/85238915
今日推荐