【线段树】 SPOJ 1716 Can you answer these queries III

第一题再加上更新操作。。。

给出n个数

询问 l- r 区间内一段连续区间的最大值

sum表示整个节点的值

maxx表示该节点的上的最大值

lsum表示该节点从左端开始的最大值

rsum表示该节点从右端开始的最大值

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <string>
#include <iostream>
#include <algorithm>
#include <sstream>
#include <cmath>
using namespace std;
#include <queue>
#include <stack>
#include <vector>
#include <deque>
#define cler(arr, val)    memset(arr, val, sizeof(arr))
#define FOR(i,a,b)  for(int i=a;i<=b;i++)
#define IN   freopen ("in.txt" , "r" , stdin);
#define OUT  freopen ("out.txt" , "w" , stdout);
typedef long long  LL;
const int MAXN = 50020;
const int MAXM = 200000;
const int INF = 0x3f3f3f3f;
const int mod = 1000000007;
const double eps= 1e-8;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
int sum[MAXN<<2],lsum[MAXN<<2],num[MAXN],rsum[MAXN<<2],maxx[MAXN<<2];
void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
    lsum[rt]=max(lsum[rt<<1],sum[rt<<1]+lsum[rt<<1|1]);
    rsum[rt]=max(rsum[rt<<1|1],sum[rt<<1|1]+rsum[rt<<1]);

    maxx[rt]=max(maxx[rt<<1],maxx[rt<<1|1]);
    maxx[rt]=max(maxx[rt],lsum[rt]);
    maxx[rt]=max(maxx[rt],rsum[rt]);
    maxx[rt]=max(maxx[rt],rsum[rt<<1]+lsum[rt<<1|1]);
}
void build(int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=lsum[rt]=rsum[rt]=maxx[rt]=num[l];
        return ;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}
int query_l(int L,int R,int l,int r,int rt)//取左节点的最大值
{
    if(L==l&&R==r) return rsum[rt];
    int m=(l+r)>>1;
    if(m<L) return query_l(L,R,rson);
    else
    {
        int s=rsum[rt<<1|1];
        int ls=sum[rt<<1|1]+query_l(L,m,lson);//sum[rt<<1|1]+rsum[rt<<1];
        return max(s,ls);
    }
}
int query_r(int L,int R,int l,int r,int rt)//取右节点的最大值
{
    if(L==l&&R==r) return lsum[rt];
    int m=(l+r)>>1;
    if(m>=R) return query_r(L,R,lson);
    else
    {
        int s=lsum[rt<<1];
        int rs=sum[rt<<1]+query_r(m+1,R,rson);//sum[rt<<1]+lsum[rt<<1|1];
        return max(s,rs);
    }
}
int query(int L,int R,int l,int r,int rt)
{
    if(L==l&&R==r)
        return maxx[rt];
    int m=(l+r)>>1;
    if(m>=R) return query(L,R,lson);
    else if(L>m) return query(L,R,rson);
    else
    {
        int ls=query(L,m,lson);
        int rs=query(m+1,R,rson);
        int ms=query_l(L,m,lson)+query_r(m+1,R,rson);//rsum[rt<<1]+lsum[rt<<1|1];
        return max(max(ls,rs),ms);
    }
}
void update(int A,int B,int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=lsum[rt]=rsum[rt]=maxx[rt]=B;
        return ;
    }
    int m=(l+r)>>1;
    if(A<=m) update(A,B,lson);
    else update(A,B,rson);
    pushup(rt);
}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
  //  freopen("out.txt", "w", stdout);
#endif
    int m,n;
    while(scanf("%d",&n)!=EOF)
    {
        for(int i=1;i<=n;i++)scanf("%d",&num[i]);
        build(1,n,1);
        scanf("%d",&m);
        for(int i=0;i<m;i++)
        {
            int ls,rs,op;
            scanf("%d%d%d",&op,&ls,&rs);
            if(op)
                printf("%d\n",query(ls,rs,1,n,1));
            else update(ls,rs,1,n,1);
        }
    }
    return 0;
}


posted @ 2014-10-18 09:07  kewowlo  阅读(111)  评论(0编辑  收藏  举报