hdu 1166(树状数组或线段树)

最新推荐文章于 2021-06-09 19:17:31 发布

转载最新推荐文章于 2021-06-09 19:17:31 发布 · 97 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Mathics/p/3681183.html

本文详细介绍了线段树和树状数组的基本概念、实现原理及应用场景，并通过具体的代码示例展示了这两种数据结构的使用方法。

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

线段树 (本题无需建树,少了很多)

#include<cstdio>
#include<cstring>
int sum[5000005],rt,data,lb,rb,n,m;
void add(int p,int l,int r,int now)//data新加的数 第p个位置 第now个子树
{
    sum[now]+=data;
    if(l==r) return;
    int mid=(l+r)/2;
    if(p<=mid) add(p,l,mid,2*now);
    else add(p,mid+1,r,2*now+1);
}
int query(int l,int r,int now)
{
    int mid=(l+r)/2;
    if(lb<=l&&rb>=r) return(sum[now]);
    if(rb<=mid) return(query(l,mid,2*now));
    else if(lb>mid) return(query(mid+1,r,2*now+1));
    else return(query(l,mid,2*now)+query(mid+1,r,2*now+1));
}
int main()
{
    int t,tt,i,j;
    char s[100];
    scanf("%d",&t);
    for(tt=1;tt<=t;tt++){
        scanf("%d",&n);
        printf("Case %d:\n",tt);
        memset(sum,0,sizeof(sum));
        for(i=1;i<=n;i++){
            scanf("%d",&data);
            add(i,1,n,1);
        }
        while(1){
            scanf("%s",&s);
            if(s[0]=='E') break; else scanf("%d%d",&i,&j);
            switch(s[0]){
                case 'A':data=j;add(i,1,n,1);break;
                case 'S':data=-j;add(i,1,n,1);break;
                case 'Q':lb=i;rb=j;printf("%d\n",query(1,n,1));break;
            }
        }
    }
    return 0;
}

简单的树状数组

先给个经典图片

#include<iostream>
#include<cstdio>
#include<cstring>
int n,a[50005];
int lowbit(int a)
{
    return(a&(-a));
}
void add(int p,int q)
{
    for(int i=q;i<=n;i+=lowbit(i))
        a[i]+=p;
}
int sum(int q)
{
    int ans=0;
    for(int i=q;i>0;i-=lowbit(i))
        ans+=a[i];
    return(ans);
}
int main()
{
    int t,tt,i,j,k;
    char s[100];
    scanf("%d",&t);
    for(tt=1;tt<=t;tt++){
        scanf("%d",&n);
        printf("Case %d:\n",tt);
        memset(a,0,sizeof(a));
        for(i=1;i<=n;i++){
            scanf("%d",&k);
            add(k,i);
        }
        while(1){
            scanf("%s",&s);
            if(s[0]=='E') break; else scanf("%d%d",&i,&j);
            switch(s[0]){
                case 'A':add(j,i);break;
                case 'S':add(-j,i);break;
                case 'Q':printf("%d\n",sum(j)-sum(i-1));break;
            }
        }
    }
    return 0;
}

转载于:https://www.cnblogs.com/Mathics/p/3681183.html