1028 字
5 分钟
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
}