1. 程式人生 > >Can you answer these queries-剪枝點更新

Can you answer these queries-剪枝點更新

題意: 段更新,但是直接更新到點會超時 ,由於求得是區間和但是更新的是區間內每一個點的sqrt顯然lazy標記已經無法完成

所以必須要進行點更新在更新總區間的和,但是可以剪枝優化,在題目所給的範圍內最多sqrt 七次就會到1之後就不在變化了所以判斷一下這個區間的和是否等於全為一時的和如果是的話直接停止更新。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define maxn 5010001
struct node
{
    ll l,r,sum;
    ll mid()
    {
        return (l+r)/2;
    }
} tree[maxn*4];
void build(ll root,ll l,ll r)
{
    tree[root].l=l;
    tree[root].r=r;
    if(l==r)
    {
        scanf("%lld",&tree[root].sum);
        return ;
    }
    build(root*2,l,tree[root].mid());
    build(root*2+1,tree[root].mid()+1,r);
    tree[root].sum=tree[root*2].sum+tree[root*2+1].sum;
}
void updata(ll root,ll l,ll r)
{
    if(tree[root].l==l&&tree[root].r==r&&tree[root].sum==(r-l+1))
        return ;
    if(tree[root].l==tree[root].r)
    {
        tree[root].sum=sqrt(tree[root].sum*1.0);
        return ;
    }
    if(r<=tree[root].mid())
        updata(root*2,l,r);
    else if(l>tree[root].mid())
        updata(root*2+1,l,r);
    else
    {
        updata(root*2,l,tree[root].mid());
        updata(root*2+1,tree[root].mid()+1,r);
    }
    tree[root].sum=tree[root*2].sum+tree[root*2+1].sum;
}
ll query(ll root,ll l,ll r)
{
    if(tree[root].l==l&&tree[root].r==r)
        return tree[root].sum;
    if(l>tree[root].mid())
        return query(root*2+1,l,r);
    else if(r<=tree[root].mid())
        return query(root*2,l,r);
    else
        return query(root*2,l,tree[root].mid())+query(root*2+1,tree[root].mid()+1,r);
}
ll n,q,x,y,z,t;
int main()
{
    while(~scanf("%lld",&n))
    {
        ++t;
        build(1,1,n);
        scanf("%lld",&q);
        printf("Case #%lld:\n",t);
        while(q--)
        {
            scanf("%lld%lld%lld",&x,&y,&z);
            if(y>z)
                swap(y,z);
            if(x==0)
                updata(1,y,z);
            else
                printf("%lld\n",query(1,y,z));
        }
        printf("\n");
    }
    return 0;
}