tjut 3577

本文介绍了一种使用懒标记优化的线段树数据结构实现方法,并通过具体代码示例展示了如何进行区间更新和查询操作,特别适用于解决区间修改与查询的问题。
#include<iostream>  
#include<string>  
#include<queue>  
#include<map>  
#include<set>  
#include<vector>  
#include<algorithm>  
#include<cstdio>  
#include<cstring>  
using namespace std;  
const int N=1000005;  
int ans[N];  
struct node  
{  
    int l,r,v,lazy;  
}node[N<<2];    //  线段树的空间大概是数组空间的4倍;  
void build(int l,int r,int numb)    //  线段树的建立;  
{  
    node[numb].l=l;  
    node[numb].r=r;  
    node[numb].v=0;  
    node[numb].lazy=0;              //  用了lazy思想,提高了效率;  
    if(l==r) return;  
    int mid=(l+r)>>1;  
    build(l,mid,numb<<1);  
    build(mid+1,r,numb<<1|1);  
}  
void PushUp(int numb)               //  往上往父节点方向更新数据;但是这里不是左右儿子的和,而是最大值,因为是站台人数;  
{  
    node[numb].v=max(node[numb<<1].v,node[numb<<1|1].v);  
}  
void PushDown(int numb)             //  向下往左右儿子方向更新数据;  
{  
    node[numb<<1].lazy+=node[numb].lazy;  
    node[numb<<1|1].lazy+=node[numb].lazy;  
    node[numb<<1].v+=node[numb].lazy;  
    node[numb<<1|1].v+=node[numb].lazy;  
    node[numb].lazy=0;              //  更新完了要清零;  
}  
void Insert(int l,int r,int numb)   //  插入更新数据;  
{  
    if(node[numb].l==l&&node[numb].r==r)    //  如果区间完全重合,则不需要再往下更新了,先保存起来,可以节约很多的时间(lazy思想)  
    {  
        node[numb].v+=1;  
        node[numb].lazy+=1;  
        return;  
    }  
    if(node[numb].lazy) PushDown(numb);     //  因为没有找到完全重合的区间,所以要先更新下一层区间;  
    int mid=(node[numb].r+node[numb].l)>>1;  
    if(l>mid) Insert(l,r,numb<<1|1);  
    else if(r<=mid) Insert(l,r,numb<<1);  
    else{  
        Insert(l,mid,numb<<1);  
        Insert(mid+1,r,numb<<1|1);  
    }  
    PushUp(numb);       //  最后还得往上返回,更新父节点区间;  
}  
int query(int l,int r,int numb)     //  查询区间l到r;  
{  
    if(node[numb].l==l&&node[numb].r==r){  
        return node[numb].v;  
    }  
    if(node[numb].lazy) PushDown(numb);     //  道理同48行;  
    int mid=(node[numb].r+node[numb].l)>>1;  
    if(l>mid) return query(l,r,numb<<1|1);  
    else if(r<=mid) return query(l,r,numb<<1);  
    else{  
        return max(query(l,mid,numb<<1),query(mid+1,r,numb<<1|1));  //  道理同28行;  
    }  
}  
int main()  
{  
    int t,Case=1,len=0,k,m,a,b;  
    scanf("%d",&t);  
    while(t--){  
        len=0;  
        memset(ans,0,sizeof(ans));  
        scanf("%d%d",&k,&m);  
        build(1,1000000,1);  
        for(int i=0;i<m;i++){  
            scanf("%d%d",&a,&b);  
            b--;                    //  这里有一个问题,就是乘客从a上车,b下车,所以乘客在车上的区间为(a,b--);  
            if(query(a,b,1)<k){     //  表示可以上车;  
                ans[len++]=i+1;  
                Insert(a,b,1);  
            }  
        }  
        printf("Case %d:\n",Case++);  
        for(int i=0; i<len; i++)    //  格式问题害我又WA了一次;  
            printf("%d ",ans[i]);  
        printf("\n\n");  
    }  
    return 0;  
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值