bzoj3038(线段树开根)

法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;
}


 

总结:

线段树在打标记时的更新当前数据那个地方需要非常关注,容易出错。

同时数据结构题对拍的重要性真的是不言而喻!对拍的话,这些细节就很容易避免了。

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值