您的位置:首页 > 产品设计 > UI/UE

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

2016-08-25 10:08 381 查看
题目大意:

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

分析:

这里开根号数会越来越小,到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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  线段树