线段树_Can you answer these queries?

http://acm.hdu.edu.cn/showproblem.php?pid=4027
题意:给一个数组序列, 数组长度为100000,现在有两种操作, 一种操作是将某一个固定区间所有数开方(向下取整),另一种操作是询问某个区间的所有数字之和。
题解:线段树:单点更新,区间求和。注意一个可以优化的地方:在query和update操作中,如果segTree[rt]==r-l+1,说明该区间内所有数都为1,开方也是1,则不必递归下去了。

#include<stdio.h>
#include<cmath>
#include<algorithm>
#include <string.h>
using namespace std;
#define maxn 100010
typedef long long LL;
LL segTree[maxn<<2];
void build(int l,int r,int rt)
{
    if(l==r)
    {
        scanf("%lld",&segTree[rt]);
        return;
    }
    int mid=l+((r-l)>>1);
    build(l,mid,rt<<1);
    build(mid+1,r,(rt<<1)|1);
    segTree[rt]=segTree[rt<<1]+segTree[(rt<<1)|1];
}
LL query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R) return segTree[rt];
    if(L<=l&&r<=R&&segTree[rt]==r-l+1) return r-l+1;
    int mid=l+((r-l)>>1);
    LL sum=0;
    if(mid>=L) sum=sum+query(L,R,l,mid,rt<<1);
    if(mid+1<=R) sum=sum+query(L,R,mid+1,r,(rt<<1)|1);
    return sum;
}
void update(int L,int R,int l,int r,int rt)
{
    if(l==r)
    {
        segTree[rt]=sqrt(segTree[rt]);
        return;
    }
    if(L<=l&&r<=R&&segTree[rt]==r-l+1) return;
    int mid=l+((r-l)>>1);
    if(L<=mid) update(L,R,l,mid,rt<<1);
    if(mid+1<=R) update(L,R,mid+1,r,(rt<<1)|1);
    segTree[rt]=segTree[rt<<1]+segTree[(rt<<1)|1];
}
int main()
{
    int n,m,L,R,op,cas=0;
    while(scanf("%d",&n)!=EOF)
    {
        build(1,n,1);
        printf("Case #%d:\n",++cas);
        scanf("%d",&m);
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&op,&L,&R);
            if(L>R) swap(L,R);
            if(op)
            {
                printf("%lld\n",query(L,R,1,n,1));
            }
            else
            {
                update(L,R,1,n,1);
            }
        }
        printf("\n");
    }
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • 待更新 线段树讲解(未读)线段树模板(未读) 模板 求区间总和 A - 敌兵布阵 HDU - 1166 题意 线段...
    染微言阅读 5,081评论 1 0
  • 主席树的作用是寻找一个序列的某个区间的第k大(小)如:给出一个序列a1,a2...an,有若干个询问,每个询问形如...
    Gitfan阅读 3,564评论 0 0
  • http://blog.csdn.net/liuledidai/article/details/9964697[h...
    Gitfan阅读 3,821评论 0 0
  • created by Dejavu(不断更新中) 概念 线段树,类似区间树,是一个完全二叉树,它在各个节点保存一条...
    ericdejavu阅读 5,124评论 0 0
  • http://acm.hdu.edu.cn/showproblem.php?pid=1525 题意:给你两个数,每...
    Gitfan阅读 4,592评论 0 0

友情链接更多精彩内容