線段樹_Can you answer these queries?

http://acm.hdu.edu.cn/showproblem.php?pid=4027
題意:給一個數(shù)組序列, 數(shù)組長度為100000,現(xiàn)在有兩種操作, 一種操作是將某一個固定區(qū)間所有數(shù)開方(向下取整),另一種操作是詢問某個區(qū)間的所有數(shù)字之和。
題解:線段樹:單點更新,區(qū)間求和。注意一個可以優(yōu)化的地方:在query和update操作中,如果segTree[rt]==r-l+1,說明該區(qū)間內(nèi)所有數(shù)都為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");
    }
}
最后編輯于
?著作權歸作者所有,轉載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內(nèi)容

  • 待更新 線段樹講解(未讀)線段樹模板(未讀) 模板 求區(qū)間總和 A - 敵兵布陣 HDU - 1166 題意 線段...
    染微言閱讀 1,353評論 1 0
  • 主席樹的作用是尋找一個序列的某個區(qū)間的第k大(小)如:給出一個序列a1,a2...an,有若干個詢問,每個詢問形如...
    Gitfan閱讀 614評論 0 0
  • http://blog.csdn.net/liuledidai/article/details/9964697[h...
    Gitfan閱讀 706評論 0 0
  • created by Dejavu(不斷更新中) 概念 線段樹,類似區(qū)間樹,是一個完全二叉樹,它在各個節(jié)點保存一條...
    ericdejavu閱讀 1,397評論 0 0
  • http://acm.hdu.edu.cn/showproblem.php?pid=1525 題意:給你兩個數(shù),每...
    Gitfan閱讀 1,084評論 0 0

友情鏈接更多精彩內(nèi)容