hdu4027_Can you answer these queries?

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;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值