题目链接:https://vjudge.net/problem/HDU-5828
参考博客:
http://www.cnblogs.com/forever97/p/hdu5828.html
http://blog.csdn.net/zzz805/article/details/52193127
区间开方不好维护,但是多次开方后区间极差会快速降低到小于等于1。
我们在区间极差大于1的时候暴力更新。
在小于等于1的时候讨论。
如果区间极差为0,那么整个区间的元素都相等,可以直接维护(区间赋值或区间更新)。
如果区间极差为1,那么整个区间只有两种相差1的值,再讨论下开方后两种值是否相等还是依然相差1。
如果开方后相等,那么就直接维护(区间赋值),否则就可以转化为区间加上一个值(区间更新)。
具体算法就是线段树维护区间和,区间最大值,区间最小值。极差可以直接算出来。
延迟标记维护区间增减。
开方操作转化成区间增减或区间赋值。
区间赋值就是一个flag标记,表示这个区间内所有的值都相等,要往下push,flag可以直接由区间最值判断。
pushdown函数写错了,少写了add(ls),add(rs)的更新,WA了好久。
自己对本题完全没有想法,还是经验不够丰富,思考确实很重要,但也要多学习别人的方法。
长代码中的小错误很难搞定,哪怕自己带数据也很难找到错误。
写代码的时候再小心一点吧,检查的时候要从多个层次考虑把,包括细节,实现,算法。
查错可以带数据,检查,对拍。
但能够避免错误是最好的。
本题是利用了区间开方可以缩小区间极差的特点。然后当区间极差比较小的时候,可以通过讨论将区间开方转化为区间增减或区间赋值。
代码
#include<stdio.h>
#include<algorithm>
#include<math.h>
#define ls (now<<1)
#define rs (ls|1)
#define m ((l+r)>>1)
using namespace std;
typedef long long ll;
const int maxn = 100010;
int n,q;
int A[maxn];
/线段树
ll tree[maxn<<2][3];
ll add[maxn<<2];
void maintain(int now)
{
tree[now][0]=tree[ls][0]+tree[rs][0];
tree[now][1]=max(tree[ls][1],tree[rs][1]);
tree[now][2]=min(tree[ls][2],tree[rs][2]);
}
void pushdown(int l,int r,int now)
{
if(tree[now][1]==tree[now][2])
{
tree[ls][0]=tree[now][1]*(m-l+1);
tree[rs][0]=tree[now][1]*(r-m);
for(int i=1;i<=2;i++) tree[ls][i]=tree[rs][i]=tree[now][i];
add[now]=0;
}
else if(add[now])
{
tree[ls][0]+=add[now]*(m-l+1);
tree[rs][0]+=add[now]*(r-m);
for(int i=1;i<=2;i++)
{
tree[ls][i]+=add[now];
tree[rs][i]+=add[now];
}
add[ls]+=add[now];
add[rs]+=add[now];
add[now]=0;
}
}
void build(int l,int r,int now)
{
add[now]=0;
if(l==r)
{
for(int i=0;i<3;i++) tree[now][i]=A[l];
return;
}
build(l,m,ls);
build(m+1,r,rs);
maintain(now);
}
void update1(int l,int r,int now,int ql,int qr,ll val)
{
if(l>qr||r<ql) return;
if(ql<=l&&r<=qr)
{
tree[now][0]+=val*(r-l+1);
for(int i=1;i<=2;i++) tree[now][i]+=val;
add[now]+=val;
return;
}
pushdown(l,r,now);
update1(l,m,ls,ql,qr,val);
update1(m+1,r,rs,ql,qr,val);
maintain(now);
}
void update2(int l,int r,int now,int ql,int qr)
{
if(l>qr||r<ql) return;
if(ql<=l&&r<=qr)
{
if(tree[now][1]==tree[now][2])
{
ll val = floor(sqrt(tree[now][1]))-tree[now][1];
tree[now][0]+=val*(r-l+1);
for(int i=1;i<=2;i++) tree[now][i]+=val;
add[now]+=val;
return;
}
if(tree[now][1]==tree[now][2]+1)
{
ll s = floor(sqrt(tree[now][2]));
ll b = floor(sqrt(tree[now][1]));
if(s==b)
{
tree[now][0]=b*(r-l+1);
tree[now][1]=tree[now][2]=b;
}
else
{
ll val = b-tree[now][1];
tree[now][0]+=val*(r-l+1);
for(int i=1;i<=2;i++) tree[now][i]+=val;
add[now]+=val;
}
return;
}
}
pushdown(l,r,now);
update2(l,m,ls,ql,qr);
update2(m+1,r,rs,ql,qr);
maintain(now);
}
ll qry(int l,int r,int now,int ql,int qr)
{
if(l>qr||r<ql) return 0;
if(ql<=l&&r<=qr) return tree[now][0];
pushdown(l,r,now);
return qry(l,m,ls,ql,qr)+qry(m+1,r,rs,ql,qr);
}
/线段树
void read()
{
scanf("%d %d",&n,&q);
for(int i=1;i<=n;i++)
scanf("%d",A+i);
}
void solve()
{
read();
build(1,n,1);
int p,l,r,x;
while(q--)
{
scanf("%d",&p);
if(p==1)
{
scanf("%d %d %d",&l,&r,&x);
update1(1,n,1,l,r,x);
}
else if(p==2)
{
scanf("%d %d",&l,&r);
update2(1,n,1,l,r);
}
else
{
scanf("%d %d",&l,&r);
printf("%lld\n",qry(1,n,1,l,r));
}
}
}
int main()
{
int T;
scanf("%d",&T);
while(T--) solve();
return 0;
}