首页 > 其他 > 详细

YBTOJ 取模问题

时间:2021-09-16 13:18:41      阅读:20      评论:0      收藏:0      [点我收藏+]

技术分享图片
技术分享图片
一道特别水的线段树基础题。

不过发生了一些有意思的事情:
我最开始没算复杂度,就直接去想怎么对区间取模操作写懒标记和pushdown。
其实因为取模不会进行太多次,而且在中间可以特判结束向下递归,因此直接单点修改即可。

#include<bits/stdc++.h>
using namespace std;
#define mid (l+r>>1)
#define lson k<<1
#define rson k<<1|1
const int maxn=1e5+40;
int n,m;
long long sum[maxn<<2],input[maxn]; 
void build(int k,int l,int r)
{
	if(l==r)
	{
		sum[k]=input[l];
		return;
	}
	build(lson,l,mid);
	build(rson,mid+1,r);
	sum[k]=sum[lson]+sum[rson];
	return;
}	
void longchange(int k,int l,int r,int x,int y,int mo)
{
	if(l==r) 
	{
		sum[k]%=mo;
		return;
	}
	if(sum[k]<mo) return;
	if(x<=mid) longchange(lson,l,mid,x,y,mo);
	if(y>=mid+1) longchange(rson,mid+1,r,x,y,mo);
	sum[k]=sum[lson]+sum[rson];
	return;
}
void dotchange(int k,int l,int r,int x,int v)
{
	if(l==r)
	{
		sum[k]=v;
		return;	
	}
	if(x<=mid) dotchange(lson,l,mid,x,v);
	else dotchange(rson,mid+1,r,x,v);
	sum[k]=sum[lson]+sum[rson];
	return; 
}
long long longquery(int k,int l,int r,int x,int y)
{
	if(x<=l && r<=y) return sum[k];
	long long res=0;
	if(x<=mid) res+=longquery(lson,l,mid,x,y);
	if(y>=mid+1) res+=longquery(rson,mid+1,r,x,y);
	return res;	
}
int main()
{
	int l,r,x,k,y,t; 
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;++i) scanf("%lld",&input[i]);
	build(1,1,n);
	for(int i=1;i<=m;++i)
	{
		scanf("%d",&t);
		if(t==1)
		{
			scanf("%d%d",&l,&r);
			long long ans=longquery(1,1,n,l,r); 
			printf("%lld\n",ans);
		}
		if(t==2)
		{
			scanf("%d%d%d",&l,&r,&x);
			longchange(1,1,n,l,r,x);	
		}
		if(t==3)
		{
			scanf("%d%d",&k,&y);
			dotchange(1,1,n,k,y);	
		}
	}	
	return 0;
}	

YBTOJ 取模问题

原文:https://www.cnblogs.com/mint-hexagram/p/15269045.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!