http://acm.hdu.edu.cn/showproblem.php?pid=4027
题目是标准的线段树,由于要对区间内的每个数进行开方操作,所以每次update都要update到叶子节点那里。
这样很明显没有体现线段树的lazy update,稍微分析也可知道必定TLE。。。
改进:
首先,题目中要求开方后向下取整。然后,由于sum最大不超过2^63,所以每个数最多开方7次,等到变成1的时候,数值就不会再变化了,也就是,如果某个区间内所有数都变成1了,那么就不需要再update了。
最后,注意,x有可能大于y的。。。若无注意到此处,会RE滴~
代码:
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N = 100000 + 10;
struct SegTree
{
int left, right;
ll sum;
bool allone; /**这里是重点,标记是否区间内全为1**/
};
SegTree st[N*3];
ll num[N];
void build(int left, int right, int idx)
{
st[idx].left = left;
st[idx].right = right;
if (left == right)
{
st[idx].sum = num[left];
if (st[idx].sum == 1)
st[idx].allone = true;
else st[idx].allone = false;
return;
}
int mid = (left + right) / 2;
build(left, mid, idx*2);
build(mid+1, right, idx*2+1);
st[idx].sum = st[idx*2].sum + st[idx*2+1].sum;
if (st[idx*2].allone && st[idx*2+1].allone)
st[idx].allone = true;
else st[idx].allone = false;
}
void update(int left, int right, int idx)
{
if (st[idx].left == st[idx].right)
{
st[idx].sum = (ll)sqrt(st[idx].sum * 1.0);
if (st[idx].sum == 1)
st[idx].allone = true;
return;
}
if (st[idx].allone && st[idx].left <= left && right <= st[idx].right)
return;
int mid = (st[idx].left + st[idx].right) / 2;
if (right <= mid)
update(left, right, idx*2);
else if (left > mid)
update(left, right, idx*2+1);
else
{
update(left, mid, idx*2);
update(mid+1, right, idx*2+1);
}
st[idx].sum = st[idx*2].sum + st[idx*2+1].sum;
if (st[idx*2].allone && st[idx*2+1].allone)
st[idx].allone = true;
}
ll query(int left, int right, int idx)
{
if (st[idx].left == left && st[idx].right == right)
return st[idx].sum;
int mid = (st[idx].left + st[idx].right) / 2;
if (left > mid)
return query(left, right, idx*2+1);
else if (right <= mid)
return query(left, right, idx*2);
else return query(left, mid, idx*2) + query(mid+1, right, idx*2+1);
}
int main()
{
// freopen("in.txt", "r", stdin);
int n, m, k = 1;
while (scanf("%d", &n) != EOF)
{
printf("Case #%d:\n", k++);
for (int i = 1; i <= n; i++)
scanf("%I64d", &num[i]);
build(1, n, 1);
int t, a, b;
scanf("%d", &m);
for (int i = 1; i <= m; i++)
{
scanf("%d%d%d", &t, &a, &b);
if (a > b)
swap(a, b);
if (t == 0)
update(a, b, 1);
else printf("%I64d\n", query(a, b, 1));
}
printf("\n");
}
return 0;
}