`
hududumo
  • 浏览: 238819 次
文章分类
社区版块
存档分类
最新评论

poj 4047 Garden 2012金华邀请赛 线段树

 
阅读更多
/*
三种操作
0 x y 把x位置的值改成y
1 x y 交换x与y位置的值
2 x y 统计[x,y]中连续k个值的最大值
我们把1-k的和值作为b[1],2-(k+1)的和值作为b[2],以此类推
那么可以建立一棵[1,n-k+1]的线段树
求连续k个和的最大值,退化为求某个区间的最值问题
线段树中有延迟操作,优化时间效率
*/
#include <cstdio>
#include <cstring>
#include <algorithm>
#include<iostream>
using namespace std;
const int N=200009;
int n,m,k,num;
int v[N],b[N];
struct Node
{
    int L,R,mid,v,lazy;//v保存线段中的最大值,lazy保存未释放的值
}node[N*5];
int create(int t,int L,int R)
{
    node[t].L=L;
    node[t].R=R;
    node[t].mid=(L+R)>>1;
    node[t].lazy=0;
    if(L==R)
    {
         node[t].v=b[L];
         return b[L];
    }
    node[t].v=max(create(t<<1,L,node[t].mid),create(t<<1|1,node[t].mid+1,R));
    return node[t].v;
}
void init()
{
    int i,j,t=0;
    scanf("%d%d%d",&n,&m,&k);
    num=n-k+1;
    for(int i=1;i<=n;i++)
    scanf("%d",&v[i]);
    memset(b,0,sizeof(b));
    for(int i=1;i<=k;i++)b[1]+=v[i];
    for(int i=2;i<=n-k+1;i++)b[i]=b[i-1]-v[i-1]+v[i+k-1];
    create(1,1,num);//以b数组建线段树
}
void down(int t)//把lazy值释放给子节点
{
    if(node[t].L==node[t].R)return ;
    node[t<<1].v+=node[t].lazy;
    node[t<<1].lazy+=node[t].lazy;
    node[t<<1|1].v+=node[t].lazy;
    node[t<<1|1].lazy+=node[t].lazy;
    node[t].lazy=0;
}
void up(int t)//子节点值改变后更新父节点的最大值
{
    if(node[t].L==node[t].R)return ;
    node[t].v=max(node[t<<1].v,node[t<<1|1].v);
}
int query(int t,int L,int R)
{
    int mid=(L+R)>>1;
    if(node[t].L==L&&node[t].R==R)
    {
        return node[t].v;
    }
    if(node[t].lazy)//有未释放的值先释放
    {
        down(t);
    }
    if(R<=node[t].mid)
    {
        return query(t<<1,L,R);
    }
    else if(L>node[t].mid)
    {
        return query(t<<1|1,L,R);
    }
    else
    {
        return max(query(t<<1,L,node[t].mid),query(t<<1|1,node[t].mid+1,R));
    }
}
void set_val(int t,int L,int R,int v)
{
    int mid=(L+R)>>1;
    if(node[t].L==L&&node[t].R==R)
    {
        node[t].v+=v;
        node[t].lazy+=v;
        return ;
    }
    if(node[t].lazy)//有未释放的值先释放
    {
        down(t);
    }
    if(R<=node[t].mid)
    {
        set_val(t<<1,L,R,v);
    }
    else if(L>node[t].mid)
    {
        set_val(t<<1|1,L,R,v);
    }
    else
    {
       set_val(t<<1,L,node[t].mid,v);
       set_val(t<<1|1,node[t].mid+1,R,v);
    }
    up(t);//子节点值改变可能会影响父节点,需要更新一遍
}
void solve()
{
    int op,a,b;
    while(m--)
    {
        scanf("%d%d%d",&op,&a,&b);
        if(op==0)
        {
            set_val(1,max(1,a-k+1),min(a,num),b-v[a]);//把a位置的值设成b,表现为线段[max(1,a-k+1),min(a,num)]加上b-v[a];
            v[a]=b;
        }
        else if(op==1)
        {
            if(a==b)continue;
            set_val(1,max(1,a-k+1),min(a,num),v[b]-v[a]);
            set_val(1,max(1,b-k+1),min(b,num),v[a]-v[b]);
            swap(v[a],v[b]);
        }
        else
        {
           printf("%d\n",query(1,a,b-k+1));
        }
    }
}
int main(){
    int T;
    scanf("%d", &T);
    while (T--)
    {
        init();
        solve();
    }
    return 0;
}

分享到:
评论

相关推荐

Global site tag (gtag.js) - Google Analytics