ZOJ Monthly, January 2018 - E Yet Another Data Structure Problem




题意:

线段树的乘和次方操作。


POINT:

学习一下lazy数组的更新就行了。 类似区间加和乘操作。 bozj 1798

mul和fang数组代表,下面的区间要^fang 然后乘mul。

mul操作就在mul数组正常更新。

fang操作就要把mul数组也更新掉。 mul=mul^fang。


#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <vector>
#include <math.h>
using namespace std;
#define LL long long
#define lt x<<1
#define rt x<<1|1
const LL maxn = 400000+55;
const LL mod = 1e9+7;
LL a[maxn];
LL sum[maxn],mul[maxn],fang[maxn],len[maxn];
void build(LL x,LL l,LL r)
{
	if(l==r){
		sum[x]=a[l];
		len[x]=1;
	}else{
		LL mid = (l+r)>>1;
		build(lt,l,mid);
		build(rt,mid+1,r);
		sum[x]=sum[lt]*sum[rt]%mod;
		len[x]=len[lt]+len[rt];
	}
	mul[x]=fang[x]=1;
}
LL qkm(LL base,LL mi)
{
	LL ans = 1;
	while(mi){
		if(mi&1) ans = ans * base %mod;
		base = base*base %mod;
		mi>>=1;
	}
	return ans;
}

void pushdown(LL x,LL l,LL r)
{
	if(mul[x]==1&&fang[x]==1) return;
	mul[lt]=qkm(mul[lt],fang[x])*mul[x]%mod;
	fang[lt]=fang[lt]*fang[x]%(mod-1);
	mul[rt]=qkm(mul[rt],fang[x])*mul[x]%mod;
	fang[rt]=fang[rt]*fang[x]%(mod-1);
	sum[lt]=qkm(sum[lt],fang[x])*qkm(mul[x],len[lt])%mod;
	sum[rt]=qkm(sum[rt],fang[x])*qkm(mul[x],len[rt])%mod;
	mul[x]=1,fang[x]=1;
}

void change1(LL x,LL l,LL r,LL ll,LL rr,LL v)//mul
{
	if(ll<=l&&rr>=r){
		sum[x]=sum[x]*qkm(v,len[x])%mod;
		mul[x]=mul[x]*v%mod;
	}else{
		pushdown(x,l,r);
		LL mid = (l+r)>>1;
		if(ll<=mid) change1(lt,l,mid,ll,rr,v);
		if(mid+1<=rr) change1(rt,mid+1,r,ll,rr,v);
		sum[x]=sum[lt]*sum[rt]%mod;
	}
}

void change2(LL x,LL l,LL r,LL ll,LL rr,LL k)//fang
{
	if(ll<=l&&rr>=r){
		sum[x]=qkm(sum[x],k)%mod;
		fang[x]=fang[x]*k%(mod-1);
		mul[x]=qkm(mul[x],k)%mod;
	}else{
		pushdown(x,l,r);
		LL mid = (l+r)>>1;
		if(ll<=mid) change2(lt,l,mid,ll,rr,k);
		if(mid+1<=rr) change2(rt,mid+1,r,ll,rr,k);
		sum[x]=sum[lt]*sum[rt]%mod;
	}
}


LL query(LL x,LL l,LL r,LL ll,LL rr)
{
	LL ans = 1;
	if(ll<=l&&rr>=r){
		ans = sum[x];
	}else{
		pushdown(x,l,r);
		LL mid = (l+r)>>1;
		if(ll<=mid) (ans*=query(lt,l,mid,ll,rr))%=mod;
		if(mid+1<=rr) (ans*=query(rt,mid+1,r,ll,rr))%=mod;
	}
	return ans%mod;
}
int main()
{
	LL T;
	scanf("%lld",&T);
	while(T--){
		LL n,q;scanf("%lld %lld",&n,&q);
		for(LL i=1;i<=n;i++) scanf("%lld",&a[i]);
		build(1,1,n);
		while(q--){
			LL op;scanf("%lld",&op);
			if(op==1){
				LL l,r,v;scanf("%lld %lld %lld",&l,&r,&v);
				change1(1,1,n,l,r,v);
			}else if(op==2){
				LL l,r,k;scanf("%lld %lld %lld",&l,&r,&k);
				change2(1,1,n,l,r,k);
			}else{
				LL l,r;scanf("%lld %lld",&l,&r);
				printf("%lld\n",query(1,1,n,l,r));
			}
		}
	}


}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值