hdu1166敌兵布阵

最新推荐文章于 2019-01-02 11:41:36 发布

ankerfeng

最新推荐文章于 2019-01-02 11:41:36 发布

阅读量442

点赞数

分类专栏：树状数组线段树STRMQ 文章标签：树状数组

本文链接：https://blog.csdn.net/imfenghui/article/details/28258557

版权

树状数组线段树STRMQ 专栏收录该内容

11 篇文章 0 订阅

订阅专栏

题目大意：

更新点，查询区间。

解题思路：

1、树状数组：

#include<stdio.h>
#include<string.h>
#define N 500000

int sum[N];
inline int lowbit(int x)
{
       return x&-x;
}
void add(int i,int x)
{
    while(i<N)
    {
        sum[i]+=x;
        i+=lowbit(i);
    }
}
int getsum(int i){
   int ans=0;
   while(i>0){
        ans+=sum[i];
        i-=lowbit(i);
   }
   return ans;
}
int main()
{
    int t,cs=0,n,i,a,b;
    char op[10];
    bool flag;
    scanf("%d",&t);
    while(t--){
        ++cs;
        memset(sum,0,sizeof(sum));
        printf("Case %d:\n",cs);
        scanf("%d",&n);
        for(i=1;i<=n;i++){
            scanf("%d",&a);
            add(i,a);
            }
        getchar();
        flag=true;
        while(flag)
        {
            scanf("%s",op);
            switch(op[0])
            {
            case 'Q':
                scanf("%d%d",&a,&b);
                printf("%d\n",getsum(b)-getsum(a-1));
                break;

            case 'A':
                    scanf("%d%d",&a,&b);
                    add(a,b);
                    break;

            case 'S':
                   scanf("%d%d",&a,&b);
                   add(a,-b);
                   break;
            default:flag=false;
                break;

         }
      }
    }
    return 0;
}

2、线段树

#include<stdio.h>
#include<string.h>
struct seg
{
    int l;
    int r;
    int sum;
}T[150011];
void build(int l,int r,int rt)
{

        T[rt].l=l;
        T[rt].r=r;
        T[rt].sum=0;
    if(l==r)
        return ;
    int mid=(l+r)/2;
    build(l,mid,2*rt);
    build(mid+1,r,2*rt+1);
}
void add(int n,int d,int rt)
{
    if(T[rt].l==T[rt].r)    {
            if(T[rt].l==d);
            T[rt].sum+=n;
            return ;
    }
    int mid=(T[rt].l+T[rt].r)>>1;
    if(d<=mid)
        add(n,d,2*rt);
    else
        add(n,d,2*rt+1);
    T[rt].sum=T[2*rt].sum+T[2*rt+1].sum;
}
int ans;
void query(int l,int r,int rt)
{
    if(T[rt].l==l&&T[rt].r==r){
        ans+=T[rt].sum;
        return ;
    }
    if(T[rt].l==l&&T[rt].r==r)
    {
        ans+=T[rt].sum;
        return ;
    }
    int mid=(T[rt].l+T[rt].r)>>1;
    if(r<=mid)
        query(l,r,2*rt);
    else if(l>mid)
        query(l,r,2*rt+1);
    else
    {
        query(l,mid,2*rt);
        query(mid+1,r,2*rt+1);
    }
}
int main()
{
    int t,cs=0,n,i,a,b;
    char op[10];
    bool flag;
    scanf("%d",&t);
    while(t--)
    {
        ++cs;
        printf("Case %d:\n",cs);
        scanf("%d",&n);
         build(1,n,1);

        for(i=1;i<=n;i++)
        {
             scanf("%d",&a);
            add(a,i,1);
        }
        getchar();
        flag=true;
        while(flag)
        {
            ans=0;
            scanf("%s",op);
            switch(op[0])
            {
            case 'Q':
                scanf("%d%d",&a,&b);
               query(a,b,1);
                printf("%d\n",ans);
                break;

            case 'A':
                    scanf("%d%d",&a,&b);
                   add(b,a,1);
                    break;

            case 'S':
                   scanf("%d%d",&a,&b);
                   add(-b,a,1);
                   break;
            default:flag=false;
                break;
         }
      }
    }
    return 0;
}