hdu5372 Segment Game 树状数组

题意:有长度为1,2,3……的若干线段,现在在x轴上进行操作(a,b)。

          当a=0时为添加操作,在[ b,b+i ]上放上一条长度为 i 的线段(当前添加操作是第 i 次添加操作)。

          当a=1时为删除操作,把第 b 次添加操作的线段删掉。

          每次添加操作时,输出该区间范围内有多少个完整的线段。

          操作数<=2*10^5,| b |<=10^9

分析:本来很麻烦,但由于线段长度是递增的,不可能出现横跨的情况。

          直接用  右端点小于等于当前区间右端点的个数 - 左端点小于当前区间左端点的个数  即可。

          由于| b |<=10^9,所以需要进行离散化。

          由于我是把左右端点一起离散化的,树状数组要开 2 * maxn。因为这个WA了好几发,都没发现。


#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <string>
#include <cmath>
#include <vector>
#include <map>
#define clr(x, y) memset(x, y, sizeof x)
using namespace std;
typedef long long LL;
const double eps=1e-8;
const int maxn=200100;
int b0[maxn];
int a[maxn],b[maxn],id[maxn];
int tx[maxn*2],ty[maxn*2];
int dgree[maxn*2];
int len;
void add(int pos,int x,int *A)
{
    while(pos <= len+10)
    {
        A[pos] += x;
        pos += pos & -pos;
    }
}
int sum(int pos,int *A)
{
    int ans=0;
    while(pos > 0)
    {
        ans += A[pos];
        pos -= pos & -pos;
    }
    return ans;
}
int main()
{
   // freopen("input.txt","r",stdin);
    int n,t=0;
    while(scanf("%d",&n)!=EOF)
    {
        t++;
        int cnt=1;
        len=0;
        memset(tx,0,sizeof(tx));
        memset(ty,0,sizeof(ty));
        for(int i=0;i<n;i++)
        {
            scanf("%d %d",&a[i],&b[i]);
            if(a[i]==0)
            {
                b0[cnt]=b[i];
                id[i]=cnt;
                dgree[len++]=b[i];
                dgree[len++]=b[i]+cnt;
                cnt++;
            }
        }
        sort(dgree,dgree+len);
        len=unique(dgree,dgree+len)-dgree;

        printf("Case #%d:\n",t);
        for(int i=0;i<n;i++)
        {
            if(a[i]==0)
            {
                int x=lower_bound(dgree,dgree+len,b[i])-dgree+1;
                int y=lower_bound(dgree,dgree+len,b[i]+id[i])-dgree+1;
                printf("%d\n",sum(y,ty)-sum(x-1,tx));
                add(x,1,tx);
                add(y,1,ty);
            }
            else
            {
                int x=lower_bound(dgree,dgree+len,b0[b[i]])-dgree+1;
                int y=lower_bound(dgree,dgree+len,b0[b[i]]+b[i])-dgree+1;
                add(x,-1,tx);
                add(y,-1,ty);
            }
        }
    }
    return 0;
}


你可能感兴趣的:(hdu5372 Segment Game 树状数组)