bzoj 3038: 上帝造题的七分钟2 线段树||hdu 4027

3038: 上帝造题的七分钟2

Time Limit: 3 Sec  Memory Limit: 128 MB
Submit: 1066  Solved: 476
[Submit][Status][Discuss]

Description

XLk觉得《上帝造题的七分钟》不太过瘾，于是有了第二部。
"第一分钟，X说，要有数列，于是便给定了一个正整数数列。
第二分钟，L说，要能修改，于是便有了对一段数中每个数都开平方(下取整)的操作。
第三分钟，k说，要能查询，于是便有了求一段数的和的操作。
第四分钟，彩虹喵说，要是noip难度，于是便有了数据范围。
第五分钟，诗人说，要有韵律，于是便有了时间限制和内存限制。
第六分钟，和雪说，要省点事，于是便有了保证运算过程中及最终结果均不超过64位有符号整数类型的表示范围的限制。
第七分钟，这道题终于造完了，然而，造题的神牛们再也不想写这道题的程序了。"
——《上帝造题的七分钟·第二部》
所以这个神圣的任务就交给你了。

Input

第一行一个整数n，代表数列中数的个数。
第二行n个正整数，表示初始状态下数列中的数。
第三行一个整数m，表示有m次操作。
接下来m行每行三个整数k,l,r，k=0表示给[l,r]中的每个数开平方(下取整)，k=1表示询问[l,r]中各个数的和。

Output

对于询问操作，每行输出一个回答。

Sample Input

10
1 2 3 4 5 6 7 8 9 10
5
0 1 10
1 1 10
1 1 5
0 5 8
1 4 8

Sample Output

19
7
6

HINT

1：对于100%的数据，1<=n<=100000，1<=l<=r<=n，数列中的数大于0，且不超过1e12。

2：数据不保证L<=R 若L>R，请自行交换L,R，谢谢！

Source

Poetize4

题意：区间开根号，优化：最大值小于1不更新；

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pi (4*atan(1.0))
const int N=5e5+10,M=4e6+10,inf=1e9+10;
ll sum[N<<2];
ll maxx[N<<2];
ll getnum(ll x)
{
    return (ll)sqrt(x);
}
void pushup(int pos)
{
    sum[pos]=sum[pos<<1]+sum[pos<<1|1];
    maxx[pos]=max(maxx[pos<<1],maxx[pos<<1|1]);
}
void build(int l,int r,int pos)
{
    if(l==r)
    {
        scanf("%lld",&sum[pos]);
        maxx[pos]=sum[pos];
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,pos<<1);
    build(mid+1,r,pos<<1|1);
    pushup(pos);
}
void update(int L,int R,int l,int r,int pos)
{
    if(l==r)
    {
        sum[pos]=getnum(sum[pos]);
        maxx[pos]=sum[pos];
        return;
    }
    int mid=(l+r)>>1;
    if(L<=mid&&maxx[pos<<1]>1)
    update(L,R,l,mid,pos<<1);
    if(R>mid&&maxx[pos<<1|1]>1)
    update(L,R,mid+1,r,pos<<1|1);
    pushup(pos);
}
ll query(int L,int R,int l,int r,int pos)
{
    if(L<=l&&r<=R)
    return sum[pos];
    int mid=(l+r)>>1;
    ll ans=0;
    if(L<=mid)
    ans+=query(L,R,l,mid,pos<<1);
    if(R>mid)
    ans+=query(L,R,mid+1,r,pos<<1|1);
    return ans;
}
int main()
{
    int n,m,i;
    scanf("%d",&n);
    build(1,n,1);
    scanf("%d",&m);
    while(m--)
    {
        int k,l,r;
        scanf("%d%d%d",&k,&l,&r);
        if(l>r)swap(l,r);
        if(k==0)
        update(l,r,1,n,1);
        else
        printf("%lld\n",query(l,r,1,n,1));
    }
    return 0;
}