BZOJ3211:花神遊歷各國(線段樹)

題面
題意:序列,區間開根,區間求和
題解:見題意

爲什麼做這題?
爐石玩多了,只會做一眼題了

#include <iostream>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>

using namespace std;
#define mmst(a, b) memset(a, b, sizeof(a))
#define mmcp(a, b) memcpy(a, b, sizeof(b))

typedef long long LL;

const int N=400400;

int n,m;
LL t[N],s[N];

void build(int ro,int l,int r)
{
    if(l==r)
    {
        scanf("%lld",&t[ro]);
        s[ro]=t[ro];
        return;
    }
    int mid=(l+r)>>1;
    build(ro<<1,l,mid);
    build((ro<<1)|1,mid+1,r);
    t[ro]=max(t[ro*2],t[ro*2+1]);
    s[ro]=s[ro*2]+s[ro*2+1];
}

void update(int ro,int l,int r,int zuo,int you)
{
    if(l==r)
    {
        s[ro]=t[ro]=sqrt(t[ro]);
        return;
    }
    int mid=(l+r)>>1;
    if(zuo<=mid&&t[ro<<1]>1)
    update(ro<<1,l,mid,zuo,you);

    if(you>mid&&t[(ro<<1)|1]>1)
    update((ro<<1)|1,mid+1,r,zuo,you);
    t[ro]=max(t[ro*2],t[ro*2+1]);
    s[ro]=s[ro*2]+s[ro*2+1];
}

LL query(int ro,int l,int r,int zuo,int you)
{
    if(l>=zuo&&r<=you)
    return s[ro];
    if(l>you||r<zuo)
    return 0;
    int mid=(l+r)>>1;
    return query(ro<<1,l,mid,zuo,you)+query((ro<<1)|1,mid+1,r,zuo,you);
}

int main()
{
    cin>>n;
    build(1,1,n);
    cin>>m;
    while(m--)
    {
        int ops,x,y;
        scanf("%d%d%d",&ops,&x,&y);
        if(ops==1)
        printf("%lld\n",query(1,1,n,x,y));
        else
        update(1,1,n,x,y);
    }

    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章