hdu 4027 Can you answer these queries? 线段树区间修改区间查询

题目大意:

对线段树有两种操作,对连续一段整体开根号,对连续一段整体求和。

分析:

这里开根号数会越来越小,到1后根号1==1,不变,可以利用这一点。某段maxv<=1后这一段就不用往下更新了,这其实只是剪枝,并没有lazy标记优化。

还有一种优化就是某一段maxv==minv,既这一段全都相等,那么开根号后值一样,可以用lazy优化。

我第一种,两种结合起来试了一下,第二次跑的比第一次还慢,不知道是数据原因还是我代码写挫了。。

附代码:

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn=2e5+5;
ll a[maxn];
ll sum[maxn*4];
ll maxv[maxn*4];
void build(int k,int l,int r){
    if(l==r){sum[k]=maxv[k]=a[l];return ;}
    int mid=l+r>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
    sum[k]=sum[k<<1]+sum[k<<1|1];
    maxv[k]=max(maxv[k<<1],maxv[k<<1|1]);
}

void update(int k,int l,int r,int a,int b){
    if(a>r||b<l)return;
    if(l==r){if(maxv[k]<=1)return ;sum[k]=maxv[k]=floor(sqrt(1.0*maxv[k]));return ;}
    if(a<=l&&r<=b){
        if(maxv[k]<=1)return;
    }
    int mid=l+r>>1;
    update(k<<1,l,mid,a,b);
    update(k<<1|1,mid+1,r,a,b);
    sum[k]=sum[k<<1]+sum[k<<1|1];
    maxv[k]=max(maxv[k<<1],maxv[k<<1|1]);
}

ll query(int k,int l,int r,int a,int b){
    if(a>r||b<l)return 0;
    if(a<=l&&r<=b)return sum[k];
    int mid=l+r>>1;
    return query(k<<1,l,mid,a,b)+query(k<<1|1,mid+1,r,a,b);
}

int main()
{
    int n;int cas=0;
    while(scanf("%d",&n)!=EOF){
        cas++;
        for(int i=1;i<=n;i++){
            scanf("%I64d",&a[i]);
        }
        build(1,1,n);
        int m;scanf("%d",&m);printf("Case #%d:\n",cas);
        for(int i=0;i<m;i++){
            int t,l,r;
            scanf("%d%d%d",&t,&l,&r);
            if(l>r)swap(l,r);
            if(t==0){
                update(1,1,n,l,r);
            }
            else{
                printf("%I64d\n",query(1,1,n,l,r));
            }
        }
        printf("\n");
    }
    return 0;
}


lazy优化:

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
ll a[maxn];
ll sum[maxn*4];
ll maxv[maxn*4];
ll minv[maxn*4];
void build(int k,int l,int r){
    if(l==r){sum[k]=maxv[k]=minv[k]=a[l];return ;}
    int mid=l+r>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
    sum[k]=sum[k<<1]+sum[k<<1|1];
    maxv[k]=max(maxv[k<<1],maxv[k<<1|1]);
    minv[k]=min(minv[k<<1],minv[k<<1|1]);
}

void update(int k,int l,int r,int a,int b){
    int mid=l+r>>1;
    if(a>r||b<l)return;
    if(l==r){if(maxv[k]<=1)return ;minv[k]=sum[k]=maxv[k]=floor(sqrt(1.0*maxv[k]));return ;}
    if(a<=l&&r<=b){
        if(maxv[k]<=1)return;
        if(maxv[k]==minv[k]){
            maxv[k]=minv[k]=sqrt(minv[k]);sum[k]=minv[k]*(r-l+1);
            return ;
        }
    }
    if(maxv[k]==minv[k]){
        maxv[k<<1]=maxv[k<<1|1]=minv[k<<1]=minv[k<<1|1]=maxv[k];
        sum[k<<1]=minv[k<<1]*(mid-l+1);
        sum[k<<1|1]=minv[k<<1|1]*(r-mid);
    }
    update(k<<1,l,mid,a,b);
    update(k<<1|1,mid+1,r,a,b);
    sum[k]=sum[k<<1]+sum[k<<1|1];
    maxv[k]=max(maxv[k<<1],maxv[k<<1|1]);
    minv[k]=min(minv[k<<1],minv[k<<1|1]);
}

ll query(int k,int l,int r,int a,int b){
    if(a>r||b<l)return 0;
    if(a<=l&&r<=b)return sum[k];
    int mid=l+r>>1;
    if(maxv[k]==minv[k]){
        maxv[k<<1]=maxv[k<<1|1]=minv[k<<1]=minv[k<<1|1]=maxv[k];
        sum[k<<1]=minv[k<<1]*(mid-l+1);
        sum[k<<1|1]=minv[k<<1|1]*(r-mid);
    }
    return query(k<<1,l,mid,a,b)+query(k<<1|1,mid+1,r,a,b);
}

int main()
{
    int n;int cas=0;
    while(scanf("%d",&n)!=EOF){
        cas++;
        for(int i=1;i<=n;i++){
            scanf("%I64d",&a[i]);
        }
        build(1,1,n);
        int m;scanf("%d",&m);printf("Case #%d:\n",cas);
        for(int i=0;i<m;i++){
            int t,l,r;
            scanf("%d%d%d",&t,&l,&r);
            if(l>r)swap(l,r);
            if(t==0){
                update(1,1,n,l,r);
            }
            else{
                printf("%I64d\n",query(1,1,n,l,r));
            }
        }
        printf("\n");
    }
    return 0;
}



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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值