洛谷P2184 贪婪大陆

目录

• $code$

题目

戳

思路

修改区间时在开始位置$s+1$表示从则例开始埋一种雷，在结束位置$e+1$表示有一种雷埋到这里就结束了。
查询时输出$[1,r]$s的和减去$[1，l-1]$的e的和。用线段树来实现。

$code$

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
using namespace std;
const int maxn=1000001;
int n,m;
struct node{
    int l,r,s,e;
}tree[maxn+1];
inline int read(){
    int x=0;bool f=0;char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')f=!f;c=getchar();}
    while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
    return f?-x:x;
}
void build(int l,int r,int now){
    tree[now].l=l,tree[now].r=r;
    if(l==r){
        tree[now].s=0;tree[now].e=0;
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,now<<1);
    build(mid+1,r,now<<1|1);
}
void a_s(int x,int now){
    if(tree[now].l==tree[now].r){
        tree[now].s+=1;
        return;
    }
    int mid=(tree[now].l+tree[now].r)>>1;
    if(x<=mid) a_s(x,now<<1);
    else a_s(x,now<<1|1);
    tree[now].s=tree[now<<1].s+tree[now<<1|1].s;
}
void a_e(int x,int now){
    if(tree[now].l==tree[now].r){
        tree[now].e+=1;
        return;
    }
    int mid=(tree[now].l+tree[now].r)>>1;
    if(x<=mid) a_e(x,now<<1);
    else a_e(x,now<<1|1);
    tree[now].e=tree[now<<1].e+tree[now<<1|1].e;
}
int ask_s(int x,int y,int now){
    if(tree[now].l>=x&&tree[now].r<=y){
        return tree[now].s;
    }
    int mid=(tree[now].l+tree[now].r)>>1,ans=0;
    if(x<=mid) ans+=ask_s(x,y,now<<1);
    if(y>mid) ans+=ask_s(x,y,now<<1|1);
    return ans;
}
int ask_e(int x,int y,int now){
    if(tree[now].l>=x&&tree[now].r<=y){
        return tree[now].e;
    }
    int mid=(tree[now].l+tree[now].r)>>1,ans=0;
    if(x<=mid) ans+=ask_e(x,y,now<<1);
    if(y>mid) ans+=ask_e(x,y,now<<1|1);
    return ans;
}

int main(){
    n=read(),m=read();
    build(1,n,1);
    int bz,x,y;
    while(m--){
        bz=read(),x=read(),y=read();
        if(bz==1) a_s(x,1),a_e(y,1);
        else printf("%d\n",ask_s(1,y,1)-ask_e(1,x-1,1));
    }
    return 0;
}