树状数组总结

最新推荐文章于 2024-12-07 20:37:41 发布

原创最新推荐文章于 2024-12-07 20:37:41 发布 · 326 阅读

0 ·

CC 4.0 BY-SA版权

模板同时被 2 个专栏收录

55 篇文章

订阅专栏

树状数组

4 篇文章

订阅专栏

单点更新区间查询 HDU 1166

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#define LL long long
#define maxn 60000
using namespace std;
int n;
LL va,sum[maxn];
char s[15];
LL getSum(int x)
{
    LL ans = 0;
    for(int i=x;i>0;i-=(i&(-i)))ans+=sum[i];
    return ans;
}
void update(int x,LL va)
{
   for(int i=x;i<=n;i+=(i&(-i)))sum[i]+=va;
}
int main()
{
    int t,i1 = 1;
    scanf("%d",&t);
    while(t--)
    {
      scanf("%d",&n);
      memset(sum,0,sizeof(sum));
      for(int i=1;i<=n;i++)
      {
          scanf("%I64d",&va);
          update(i,va);
      }
      printf("Case %d:\n",i1);
      i1++;
      while(scanf("%s",s))
      {
          int a,b;
          if(s[0]=='E')break;
          else{
            scanf("%d%d",&a,&b);
            if(s[0]=='Q')printf("%I64d\n",getSum(b)-getSum(a-1));
            else if(s[0]=='S')update(a,-b);
            else update(a,b);
          }
      }
      system("pause");
    }
    return 0;
}

HDU 1541 键值和下标反转的题目

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define maxn 36000
using namespace std;
int sum[maxn],n,ans1[maxn];
int getSum(int x)
{
    int ans = 0;
    for(;x;x-=(x&(-x)))ans+=sum[x];
    return ans;
}
void update(int x)
{
     for(;x<=maxn;x+=(x&(-x)))sum[x]+=1;
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        memset(sum,0,sizeof(sum));
        memset(ans1,0,sizeof(ans1));
        int x,y;
        for(int i=0;i<n;i++)
        {
            scanf("%d %d",&x,&y);
            x++;
            ans1[getSum(x)]++;
            update(x);
        }
        for(int i=0;i<n;i++)printf("%d\n",ans1[i]);
    }
    return 0;
}

二维的单点的更新 poj 1195

#include<cstdio>
#include<algorithm>
#include<iostream>
#define maxn 1300
#define LL long long
using namespace std;
int sum[maxn][maxn],n,a,b,c,d,flag;
void update(int x,int y,int va)
{
    for(int i=x;i<=n;i+=(i&(-i)))
        for(int j=y;j<=n;j+=(j&(-j)))
            sum[i][j]+=va;
}
int getSum(int x,int y)
{
    LL ans = 0;
    for(int i=x;i;i-=(i&(-i)))
        for(int j=y;j;j-=(j&(-j)))ans+=sum[i][j];
    return ans;
}
int main()
{
    while(scanf("%d",&flag)!=EOF)
    {
        if(flag==0){
            scanf("%d",&n);
            for(int i=1;i<=n;i++)
                for(int j=1;j<=n;j++)sum[i][j] = 0;
        }
        else if(flag==1)
        {
            scanf("%d%d%d",&a,&b,&c);
            a++;
            b++;
            update(a,b,c);
        }
        else if(flag==2)
        {
            scanf("%d%d%d%d",&a,&b,&c,&d);
            a++;b++;c++;d++;
            printf("%d\n",getSum(c,d)-getSum(a-1,d)-getSum(c,b-1)+getSum(a-1,b-1));
        }
        else break;
    }
    return 0;
}

hdu 2852 与二分相结合使用的树状数组查询

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define maxn 100005
using namespace std;
int sum[maxn],m,flag,cnt[maxn];
int getSum(int x)
{
    int ans = 0;
    for(;x;x-=(x&(-x)))ans+=sum[x];
    return ans;
}
void update(int x,int va)
{
     for(;x<maxn;x+=(x&(-x)))sum[x]+=va;
}
void binSearch(int L,int am)
{
    int R = maxn-1,ans = - 1,ll = getSum(L-1);
    while(L<=R)
    {
        int mid = (L+R)/2;
        int pre = getSum(mid)-ll;
        if(pre<am)L = mid+1;
        else {ans = mid;R = mid-1;}
    }
    if(ans==-1)printf("Not Find!\n");
    else printf("%d\n",ans);

}
int main()
{
    while(scanf("%d",&m)!=EOF)
    {
        memset(sum,0,sizeof(sum));
        memset(cnt,0,sizeof(cnt));
        while(m--)
        {
           int a,b;
           scanf("%d",&flag);
           if(flag==0)
           {
               scanf("%d",&a);
               update(a,1);
               cnt[a]++;
           }
           else if(flag==1)
           {
                scanf("%d",&a);
                if(!cnt[a])printf("No Elment!\n");
                else {
                    cnt[a]--;
                    update(a,-1);
                }
           }
           else{
             scanf("%d %d",&a,&b);
             binSearch(a+1,b);
           }
        }
    }
    return 0;
}

hdu 3584 多维的区间更新单点查询

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
int sum[105][105][105],n,m,flag,cnt[105][105][105];
int getSum(int x,int y,int z)
{
    int ans = 0;
    for(int i=x;i;i-=(i&-i))
         for(int j=y;j;j-=(j&-j))
             for(int k=z;k;k-=(k&-k))ans+=sum[i][j][k];
    return ans%2;
}
void update(int x,int y,int z)
{
     for(int i=x;i<=n;i+=(i&-i))
         for(int j=y;j<=n;j+=(j&-j))
             for(int k=z;k<=n;k+=(k&-k))sum[i][j][k]++;
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                for(int k=1;k<=n;k++)
                {
                  sum[i][j][k] = 0,
                  cnt[i][j][k] = 0;
                }
        while(m--)
        {
            scanf("%d",&flag);
            if(flag){
                int x1,y1,z1,x2,y2,z2;
                scanf("%d%d%d%d%d%d",&x1,&y1,&z1,&x2,&y2,&z2);
                update(x1,y1,z1);
                update(x2+1,y1,z1);
                update(x1,y2+1,z1);
                update(x1,y1,z2+1);
                update(x2+1,y2+1,z1);
                update(x2+1,y1,z2+1);
                update(x1,y2+1,z2+1);
                update(x2+1,y2+1,z2+1);
            }
            else{
                int x,y,z;
                scanf("%d%d%d",&x,&y,&z);
                printf("%d\n",getSum(x,y,z));
            }
        }
    }
    return 0;
}