法1,因为最多不会开根6次,所以开根,直接搞
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<algorithm>
using namespace std;
typedef long long ll;
inline ll read()
{
ll ans,f=1;char ch;
while ((ch=getchar())<'0'||ch>'9') if (ch=='-') f=-1;ans=ch-'0';
while ((ch=getchar())>='0'&&ch<='9') ans=ans*10+ch-'0';
return ans*f;
}
int n;
struct aa
{
int l,r;
ll mx,sum;
}a[100025*4];
void up(int i)
{
a[i].mx=max(a[i<<1].mx,a[i<<1|1].mx);
a[i].sum=a[i<<1].sum+a[i<<1|1].sum;
}
void build(int i,int l,int r)
{
a[i].l=l;a[i].r=r;
if(l==r)
{
a[i].mx=a[i].sum=read();
return ;
}
int mid=(l+r)>>1;
build(i<<1,l,mid);
build(i<<1|1,mid+1,r);
up(i);
}
void updata(int i,int l,int r)
{
if (a[i].mx==1) return ;
if (a[i].l==a[i].r) {a[i].mx=a[i].sum=sqrt(a[i].mx);return;}
int mid=(a[i].l+a[i].r)>>1;
if(r<=mid) updata(i<<1,l,r);
else if (l>mid) updata(i<<1|1,l,r);
else updata(i<<1,l,mid),updata(i<<1|1,mid+1,r);
up(i);
}
ll query(int i,int l,int r)
{
if (a[i].mx==1) return (r-l+1);
if (a[i].l==a[i].r) return a[i].sum;
int mid=(a[i].l+a[i].r)>>1;
if(r<=mid) return query(i<<1,l,r);
else if(l>mid) return query(i<<1|1,l,r);
return query(i<<1,l,mid)+query(i<<1|1,mid+1,r);
}
int main()
{
n=read();
build(1,1,n);
int l,r,m=read(),op;
while (m--)
{
op=read(),l=read(),r=read();
if (l>r) swap(l,r);
if (op==0) updata(1,l,r);
else printf("%lld\n",query(1,l,r));
}
return 0;
}
方法2(我自己的):
也是在最多不超过6次的基础上想的,存一个sum【0....6】表示区间开i次根的和,发现这样是可以合并的,并且也仅仅是常数的复杂度。
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<algorithm>
using namespace std;
typedef long long ll;
inline ll read()
{
ll ans,f=1;char ch;
while ((ch=getchar())<'0'||ch>'9') if (ch=='-') f=-1;ans=ch-'0';
while ((ch=getchar())>='0'&&ch<='9') ans=ans*10+ch-'0';
return ans*f;
}
int n;
struct aa
{
int l,r,id;
ll sum[7];
}a[100005*4];
void up(int i)
{
for (int j=0;j<=6;j++)
a[i].sum[j]=(a[i<<1].l ? a[i<<1].sum[j]:0)+
(a[i<<1|1].l ? a[i<<1|1].sum[j]:0);
}
void down(int i)
{
int k=a[i].id,l=i<<1,r=i<<1|1;
if (k==0) return ;
for (int j=0;j<=6;j++) a[l].sum[j]=(j+k<=6 ? a[l].sum[j+k]:a[l].r-a[l].l+1),
a[r].sum[j]=(j+k<=6 ? a[r].sum[j+k]:a[r].r-a[r].l+1);
a[l].id+=k;a[r].id+=k;
a[i].id=0;
}
void build(int i,int l,int r)
{
a[i].l=l;a[i].r=r;
if(l==r)
{
a[i].sum[0]=read();
for (int j=1;j<=6;j++) a[i].sum[j]=sqrt(a[i].sum[j-1]);
return ;
}
int mid=(l+r)>>1;
build(i<<1,l,mid);
build(i<<1|1,mid+1,r);
up(i);
}
void updata(int i,int l,int r)
{
if (a[i].l==l&&a[i].r==r)
{
a[i].id++;
for (int j=0;j<=6;j++) a[i].sum[j]=(j+1<=6 ? a[i].sum[j+1]:a[i].r-a[i].l+1);//就应该只有是在这里wa的,原因是每updata一次,只是区间开一次跟,所以sum只需用整体移一位就好,这里当时搞错了。结果移了a【i】.id次,但是之前其实已经移了啊,再移就多了。细节必须注意,对拍德海找细节最重要的方式。
return ;
}
down(i);
int mid=(a[i].l+a[i].r)>>1;
if(r<=mid) updata(i<<1,l,r);
else if (l>mid) updata(i<<1|1,l,r);
else updata(i<<1,l,mid),updata(i<<1|1,mid+1,r);
up(i);
}
ll query(int i,int l,int r)
{
if (a[i].l==l&&a[i].r==r) return a[i].sum[0];
down(i);
int mid=(a[i].l+a[i].r)>>1;
if(r<=mid) return query(i<<1,l,r);
else if(l>mid) return query(i<<1|1,l,r);
return query(i<<1,l,mid)+query(i<<1|1,mid+1,r);
}
int main()
{
n=read();
build(1,1,n);
int l,r,m=read(),op;
while(m--)
{
op=read(),l=read(),r=read();
if (l>r) swap(l,r);
if (op==0) updata(1,l,r);
else printf("%lld\n",query(1,l,r));
}
return 0;
}
总结:
线段树在打标记时的更新当前数据那个地方需要非常关注,容易出错。
同时数据结构题对拍的重要性真的是不言而喻!对拍的话,这些细节就很容易避免了。