[kuangbin带你飞]专题七 线段树 A HDU 1166

来源:互联网 发布:腾讯qq软件下载 编辑:程序博客网 时间:2024/06/02 05:29

题目地址:https://vjudge.net/contest/66989#problem/A

思路:这是我写的第一道线段树的题目,参考了一下其他大牛的代码,学习了一波。

AC代码:

#include<bits/stdc++.h>using namespace std;const int maxn=50000+5;int t;struct seg{    int l,r,n;}T[maxn*4];void build(int l,int r,int k)//建树{    int mid;    if(l==r)//只有一个元素,节点记录该单元素    {        T[k].l=l;        T[k].r=r;        T[k].n=0;        return;    }    mid=(l+r)/2;    T[k].l=l;    T[k].r=r;    T[k].n=0;    build(l,mid,2*k);//递归构造左子树    build(mid+1,r,2*k+1);//递归构造右子树}void insert(int n,int d,int k)//更新{    int mid;    if(T[k].l==T[k].r && T[k].l==d)    {        T[k].n+=n;        return ;    }    mid=(T[k].l+T[k].r)>>1;    if(d<=mid)        insert(n,d,2*k);//在左子树    else        insert(n,d,2*k+1);    T[k].n=T[2*k].n+T[2*k+1].n;}int ans;void search(int l,int r,int k){    int mid;    if(T[k].l==l && T[k].r==r)    {        ans+=T[k].n;        return;    }    mid=(T[k].l+T[k].r)>>1;    if(r<=mid)        search(l,r,2*k);    else if(l>mid)        search(l,r,2*k+1);    else    {         search(l,mid,2*k);         search(mid+1,r,2*k+1);    }}int main(){    scanf("%d",&t);    int n;    char str[11];    for(int casei=1;casei<=t;casei++)    {        scanf("%d",&n);        build(1,n,1);        for(int i=1;i<=n;i++)        {            int temp;            scanf("%d",&temp);            insert(temp,i,1);        }        printf("Case %d:\n",casei);        while(scanf("%s",str),strcmp(str,"End"))        {            int a,b;            scanf("%d%d",&a,&b);            if(strcmp(str,"Add")==0) insert(b,a,1);            else if(strcmp(str,"Sub")==0) insert(-b,a,1);            else            {                ans=0;                search(a,b,1);                printf("%d\n",ans);            }        }    }}


0 0