SOJ 2745 树状数组 TOJ 3505 树状数组-优快云博客

本文详细介绍了树状数组的概念、基本操作以及在SOJ2745和TOJ3505两个具体问题中的应用。通过实例演示了如何利用树状数组进行区间求和与区间修改操作。

这两题都是最裸的树状数组了。。记得下标从1开始哦。

SOJ 2745

#include<stdio.h>
#include<string.h>

int a[100010];
int b[100010];
int n;
int lowbit(int x)
{
    return x&(-x);
}

void add(int pos,int num)
{
   while(pos<=n)
   {
      b[pos]+=num;
      pos+=lowbit(pos);
   }
}

int sum(int end)
{
    int s=0;
    while(end>0)
    {
       s+=b[end];
       end-=lowbit(end);
    }
    return s;
}

int main()
{
    int q;
    while(scanf("%d%d",&n,&q)==2)
    {
         int i,j;
         for(i=1;i<=n;i++)
         {
            a[i]=0;
            b[i]=0;
         }
         int f,x,y;
         while(q--)
         {
           scanf("%d%d%d",&f,&x,&y);
           if(f==0)
           printf("%d\n",sum(y+1)-sum(x));
           else
           add(x+1,y);
         }
    }
    return 0;
}

TOJ 3505

这题注意一点，每次拿走的都不能超过本身的，值不能为负

#include<stdio.h>
#include<string.h>

int a[100010];
int b[100010];
int n;
int lowbit(int x)
{
    return x&-x;
}

void add(int pos,int num)
{
     a[pos]+=num;
     while(pos<=n)
     {
        b[pos]+=num;
        pos+=lowbit(pos);
     }
}

int sum(int end)
{
    int s=0;
    while(end>0)
    {
      s+=b[end];
      end-=lowbit(end);
    }
    return s;
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        int i;
        char s[10];
        memset(b,0,sizeof(b));
        memset(a,0,sizeof(a));
        int t;
        for(i=1;i<=n;i++)
        {
            scanf("%d",&t);
            add(i,t);
        }   
        int m;
        scanf("%d",&m);
        while(m--)
        {
            scanf("%s",s);
            int x,y;
            scanf("%d%d",&x,&y);   
            if(strcmp(s,"Inquire")==0)
               printf("%d\n",sum(y)-sum(x-1));
            else if(strcmp(s,"Add")==0)
               add(x,y);
            else if(strcmp(s,"Delete")==0)
            {
               if(a[x]>=y)
               add(x,-y);
               else
               add(x,-a[x]);
            }
        }
    }
    return 0;
}