线段树2求调,样例都没过
查看原帖
线段树2求调,样例都没过
389729
BantM楼主2022/11/16 21:40

求各路大佬大神小小的看一下MnZn的代码,就不求修改了,说下哪里错了也行

这里是带取模的(应该不是取模的问题)

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define mp %p
const int maxn=1e5+1;
ll x,y,k,n,m,op,p;
ll ans[4*maxn],a[maxn],add[4*maxn],mul[4*maxn];
inline ll ls(ll a){
	return a*2;
}
inline ll rs(ll a){
	return a*2+1;
}
inline void pushup(ll p){
	ans[p]=(ans[rs(p)] mp+ans[ls(p)] mp) mp;
} 
void build(ll l,ll r,ll p){
	add[p]=0;
	mul[p]=1;
	if(l==r){
		ans[p]=a[l] mp;
		return;
	}
	ll mid=(l+r)/2;
	build(l,mid,ls(p));
	build(mid+1,r,rs(p));
	pushup(p);
}
void f(ll l,ll r,ll p,ll addk,ll mulk){
	add[p]=(add[p] mp*mulk mp+addk mp)mp;
	mul[p]=(mul[p] mp*mulk mp)mp;
	ans[p]=(ans[p] mp*mulk mp+addk mp *(r-l+1)mp)mp;
}
void pushdown(ll l,ll r,ll p){
	ll mid=(l+r)/2;
	f(l,mid,ls(p),add[p],mul[p]);
	f(mid+1,r,rs(p),add[p],mul[p]);
	add[p]=0;
	mul[p]=1;
}
void ad(ll nl,ll nr,ll l,ll r,ll p,ll k){
	if(nl<=l&&r<=nr){
		ans[p]=(ans[p]+k*(r-l+1))mp;
		add[p]=(add[p]+k)mp;
		return;
	}
	pushdown(l,r,p);
	ll mid=(l+r)/2;
	if(mid>=nl)ad(nl,nr,l,mid,ls(p),k);
	if(mid<nr)ad(nl,nr,mid+1,r,rs(p),k);
	pushup(p);
}
void mu(ll nl,ll nr,ll l,ll r,ll p,ll k){
	if(nl<=l&&r<=nr){
		ans[p]=(ans[p]*k)mp;
		add[p]=(add[p]*k)mp;
		mul[p]=(mul[p]*k)mp;
		return;
	}
	pushdown(l,r,p);
	ll mid=(l+r)/2;
	if(mid>=nl)mu(nl,nr,l,mid,ls(p),k);
	if(mid<nr)mu(nl,nr,mid+1,r,rs(p),k);
	pushup(p);
}
ll query(ll nl,ll nr,ll l,ll r,ll p){
	if(nl<=l&&r<=nr){
		return ans[p];
	}
	pushdown(l,r,p);
	ll mid=(l+r)/2;
	ll res=0;
	if(mid>=nl)res=(res+query(nl,nr,l,mid,ls(p)))mp;
	if(mid<nr)res=(res+query(nl,nr,mid+1,r,rs(p)))mp;
	cout<<res<<endl;
	return res;
}
int main(){
	cin>>n>>m>>p;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	build(1,n,1);
	while(m--){
		cin>>op;
		if(op==1){
			cin>>x>>y>>k;
			ad(x,y,1,n,1,k);
		}
		else if(op==2){
			cin>>x>>y>>k;
			mu(x,y,1,n,1,k);
		} 
		else{
			cin>>x>>y;
			cout<<query(x,y,1,n,1)<<endl;
		}
	}
	return 0;
} 

因为取模太影响可读性了,这里放一份无取模的

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define mp mod p
const int maxn=1e5+1;
ll x,y,k,n,m,op,p;
ll ans[4*maxn],a[maxn],add[4*maxn],mul[4*maxn];
inline ll ls(ll a){
	return a*2;
}
inline ll rs(ll a){
	return a*2+1;
}
inline void pushup(ll p){
	ans[p]=ans[rs(p)]+ans[ls(p)];
} 
void build(ll l,ll r,ll p){
	add[p]=0;
	mul[p]=1;
	if(l==r){
		ans[p]=a[l];
		return;
	}
	ll mid=(l+r)/2;
	build(l,mid,ls(p));
	build(mid+1,r,rs(p));
	pushup(p);
}
void f(ll l,ll r,ll p,ll addk,ll mulk){
	add[p]=add[p]*mulk+addk;
	mul[p]=mul[p]*mulk;
	ans[p]=ans[p]*mulk+addk*(r-l+1);
}
void pushdown(ll l,ll r,ll p){
	ll mid=(l+r)/2;
	f(l,mid,ls(p),add[p],mul[p]);
	f(mid+1,r,rs(p),add[p],mul[p]);
	add[p]=0;
	mul[p]=1;
}
void ad(ll nl,ll nr,ll l,ll r,ll p,ll k){
	if(nl<=l&&r<=nr){
		ans[p]+=k*(r-l+1);
		add[p]+=k;
		return;
	}
	pushdown(l,r,p);
	ll mid=(l+r)/2;
	if(mid>=nl)ad(nl,nr,l,mid,ls(p),k);
	if(mid<nr)ad(nl,nr,mid+1,r,rs(p),k);
	pushup(p);
}
void mu(ll nl,ll nr,ll l,ll r,ll p,ll k){
	if(nl<=l&&r<=nr){
		ans[p]=ans[p]*k;
		add[p]=add[p]*k;
		mul[p]=mul[p]*k;
		return;
	}
	pushdown(l,r,p);
	ll mid=(l+r)/2;
	if(mid>=nl)mu(nl,nr,l,mid,ls(p),k);
	if(mid<nr)mu(nl,nr,mid+1,r,rs(p),k);
	pushup(p);
}
ll query(ll nl,ll nr,ll l,ll r,ll p){
	ll res=0;
	if(nl<=l&&r<=nr){
		return ans[p];
	}
	pushdown(l,r,p);
	ll mid=(l+r)/2;
	if(mid>=nl)res+=query(nl,nr,l,mid,ls(p));
	if(mid<nr)res+=query(nl,nr,mid+1,r,rs(p));
	return res;
}
int main(){
	cin>>n>>m>>p;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	build(1,n,1);
	while(m--){
		cin>>op;
		if(op==1){
			cin>>x>>y>>k;
			ad(x,y,1,n,1,k);
		}
		else if(op==2){
			cin>>x>>y>>k;
			mu(x,y,1,n,1,k);
		} 
		else{
			cin>>x>>y;
			cout<<query(x,y,1,n,1)<<endl;
		}
	}
} 
2022/11/16 21:40
加载中...