BZOJ 3038 上帝造题的七分钟2 线段树

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,谢谢!

题解

线段树 去修改就可以了
区间为1后就不用改了
很暴力的

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
#define LL long long
#define lch l, m, rt << 1
#define rch m + 1, r, rt << 1 | 1

const int N = 100005;
LL Min[N << 2], Max[N << 2], Sum[N << 2];
int Sqr[N << 2];
void Pushup(int rt)
{
    Min[rt] = min(Min[rt << 1], Min[rt << 1 | 1]);
    Max[rt] = max(Max[rt << 1], Max[rt << 1 | 1]);
    if (Sum[rt << 1] != -1 && Sum[rt << 1 | 1] != -1)
        Sum[rt] = Sum[rt << 1] + Sum[rt << 1 | 1];
    else
        Sum[rt] = -1;
}
void Pushdown(int rt, int s)
{
    if (Sqr[rt])
    {
        Sqr[rt << 1] += Sqr[rt];
        Sqr[rt << 1 | 1] += Sqr[rt];
        Sum[rt << 1] = Sum[rt << 1 | 1] = -1;
        if (Max[rt << 1] > 1)
            for (int i = 1; i <= Sqr[rt]; i++)
            {
                Max[rt << 1] = sqrt(Max[rt << 1]);
                Min[rt << 1] = sqrt(Min[rt << 1]);
                if (Max[rt << 1] == 1)
                    break;
            }
        if (Max[rt << 1 | 1] > 1)
            for (int i = 1; i <= Sqr[rt]; i++)
            {
                Max[rt << 1 | 1] = sqrt(Max[rt << 1 | 1]);
                Min[rt << 1 | 1] = sqrt(Min[rt << 1 | 1]);
                if (Max[rt << 1 | 1] == 1)
                    break;
            }
        if (Max[rt << 1 | 1] == Min[rt << 1 | 1])
            Sum[rt << 1 | 1] = Max[rt << 1 | 1] * (s >> 1);
        if (Max[rt << 1] == Min[rt << 1])
            Sum[rt << 1] = Max[rt << 1] * (s - (s >> 1));
        Sqr[rt] = 0;
    }
}
void buildtree(int l, int r, int rt)
{
    if (l == r)
    {
        scanf("%lld", &Min[rt]);
        Sum[rt] = Max[rt] = Min[rt];
        return;
    }
    int m = l + r >> 1;
    buildtree(lch);
    buildtree(rch);
    Pushup(rt);
}
void Update(int L, int R, int l, int r, int rt)
{
    if (L <= l && R >= r)
    {
        Sqr[rt] += 1;
        Max[rt] = sqrt(Max[rt]), Min[rt] = sqrt(Min[rt]);
        Sum[rt] = -1;
        if (Max[rt] == Min[rt])
            Sum[rt] = Max[rt] * (r - l + 1);
        return;
    }
    Sum[rt] = -1;
    Pushdown(rt, r - l + 1);
    int m = l + r >> 1;
    if (L <= m)
        Update(L, R, lch);
    if (R > m)
        Update(L, R, rch);
    Pushup(rt);
}
LL Query(int L, int R, int l, int r, int rt)
{
    if (L <= l && R >= r)
    {
        if (Max[rt] == Min[rt])
        {
            return Sum[rt] = Max[rt] * (r - l + 1);
        }
        if (Sum[rt] != -1)
            return Sum[rt];
    }
    Pushdown(rt, r - l + 1);
    LL ans = 0;
    int m = l + r >> 1;
    if (L <= m)
        ans += Query(L, R, lch);
    if (R > m)
        ans += Query(L, R, rch);
    return ans;
    Pushup(rt);
}
void print(int l, int r, int rt)
{
    printf("l=%d,r=%d,rt=%d,Sum[rt]=%lld\n", l, r, rt, Sum[rt]);
    if (l == r)
        return;
    int m = l + r >> 1;
    print(lch);
    print(rch);
}
int main()
{
    int n, m;
    //freopen("god.in", "r", stdin);
   // freopen("god.out", "w", stdout);
    scanf("%d", &n);
    buildtree(1, n, 1);
    scanf("%d", &m);
    int op, l, r;
    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);
        }
    }
#ifdef Mine
    //while (1)
        ;
#endif
}
本文作者 : NekoMio
知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议进行许可。
本文链接 : https://www.nekomio.com/2017/07/13/34/
上一篇
下一篇