树状数组

最新推荐文章于 2025-08-11 13:38:24 发布

转载最新推荐文章于 2025-08-11 13:38:24 发布 · 48 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/wen-ge/archive/2012/10/01/2709814.html

文章标签：

#php

本文介绍了一种使用树状数组实现的数据结构，并结合二分查找解决特定问题的方法。通过对树状数组进行操作（如更新和求和），以及利用二分查找定位目标元素，实现了对数据集合的有效管理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

View Code

#include<stdio.h>
#include<string.h>
#define N 100005
#define lowbit(x) (x)&(-x)
int x,y,c[N];
void update(int i,int k)
{
    while(i<=N)
    {
        c[i]+=k;
        i+=lowbit(i);
    }
}
int getsum(int i)
{
    int sum=0;
    while(i>0)
    {
        sum+=c[i];
        i-=lowbit(i);
    }
    return sum;
}
void find(int x)
{
    int l=x+1,r=N,mid,temp,t;
    temp=getsum(x);
    if(getsum(N)-temp<y) {printf("Not Find!\n");return ;}
    while(l<r)
    {
        mid=(l+r)>>1;
        t=getsum(mid);
        if(t==temp+y) r=mid;
        else if(t>temp+y) r=mid;
        else l=mid+1;
    }
    printf("%d\n",r);
    return ;
}
int main()
{
    int m,i,p;
    while(~scanf("%d",&m))
    {
      memset(c,0,sizeof(c));
      for(i=0;i<m;i++)
      {
        scanf("%d",&p);
        if(!p)
        {
            scanf("%d",&x);
            update(x,1);
        }
        else if(p==1)
        {
            scanf("%d",&x);
            if(getsum(x)-getsum(x-1)) update(x,-1);
            else printf("No Elment!\n");
        }
        else 
        {
            scanf("%d%d",&x,&y);
            find(x);
        }
      }        
    }
    return 0;
}