两颗线段树

最新推荐文章于 2022-09-13 21:02:10 发布

Grit_ICPC

最新推荐文章于 2022-09-13 21:02:10 发布

阅读量426

点赞数

分类专栏：线段树文章标签：线段树

本文链接：https://blog.csdn.net/Grit_ICPC/article/details/50935923

版权

线段树专栏收录该内容

9 篇文章 0 订阅

订阅专栏

大意：有n个操作(n<=10000),接下来有n行,分别为k,m，k==0为在m处插入一条从m到m+i的线段(i为第i次插入)。k==1为删除第m次的线段插入。问插入线段能完全的包含多少完整的线段。

思路：用1-m+i内有的端点数-1-m内的端点数=包含的线段数目。分别用两棵树来保存点的数目。

#include<algorithm>
#include<cmath>
#include<queue>
#include<map>
#define ll long long
#define inf 0x3f3f3f3f
#include <vector>
#include<cstdio>
#include<cstring>
#include<string>
#define ls l,mid,rt<<1
#define rs mid+1,r,rt<<1|1
using namespace std;

struct node{
    int a,b;
    bool vis;
}q[100100];
int sum[100100*5],sum1[100010*5];

void Push(int rt){
    sum[rt] = sum[ rt<<1]+ sum[rt<<1|1];
}
void Push1(int rt){
    sum1[rt] = sum1[ rt<<1]+ sum1[rt<<1|1];
}

int Query(int a,int b,int l,int r,int rt){
    if(a <= l&&b >= r){
        return sum[rt];
    }
    int mid=(l + r) / 2;
    int ans=0;
    if(a <= mid)
        ans += Query(a,b,ls);
    if(mid < b)
        ans += Query(a,b,rs);
    return ans;
}

int Query1(int a,int b,int l,int r,int rt){
    if(a <= l&&b >= r){
        return sum1[rt];
    }
   int   ans = 0;
    int mid = (l + r) / 2;
    if(a <= mid)
        ans += Query1(a,b,ls);
    if(b > mid)
        ans += Query1(a,b,rs);
    return ans;
}

void add(int a,int b,int l,int r,int rt){
    if(l == r){
        sum[rt] += b;
        return ;
    }
    int mid = (l + r) / 2;
    if(a <= mid ){
        add(a,b,ls);
    }
    else
        add(a,b,rs);
    Push(rt);
}

void add1(int a,int b,int l,int r,int rt){
    if(l == r){
        sum1[rt] += b;
        return ;
    }
    int mid = (l + r) / 2;
    if(a <= mid ){
        add1(a,b,ls);
    }
    else
        add1(a,b,rs);
    Push1(rt);
}

int main(){
    int n,m,i,j,k,s;
    while(~scanf("%d",&n)){
        s = 1;
        memset(q,false,sizeof(q));
        memset(sum,0,sizeof(sum));
        memset(sum1,0,sizeof(sum1));

        for(i = 0;i < n;++ i){
            scanf("%d%d",&k,&m);
            if(!k ){
                q[s].a = m+1;
                q[s].b = m + s + 1;
                q[s].vis = true;
                ++ s;
                printf("%d\n",Query1(1,m + s + 1,1,20001,1 ) - Query(1,m,1,20001,1 ));

                add(m + 1,1,1,20001,1);
                add1(m + 1 + s,1,1,20001,1 );
            }
            else{
                if(q[m].vis){
                    add(q[m].a,-1,1,20001,1);
                    add1(q[m].b,-1,1,20001,1);
                    q[m].vis=false;
                }
            }
        }
    }
    return 0;
}