JZOJ 3494. 【NOIP2013模拟联考13】线段(segment)

题目

数轴上有很多单位线段,一开始时所有单位线段的权值都是 \(1\)。有两种操作,第一种操作将某一区间内的单位线段权值乘以 \(w\),第二种操作将某一区间内的单位线段权值取 \(w\) 次幂。并且你还需要回答一些询问,每个询问需要求出某一区间的单位线段权值之积。由于答案可能很大,你只需要求出答案 \(mod (10^9+7)\) 的值。
说明:\(n\) 个点只有 \(n-1\) 条线段。

分析

线段树懒标记基本操作
幂运算优先,然后乘法运算
对于 \([-1^9,1^9]\) 的操作区间,直接动态开店就好了
离散化随你

\(Code\)

#include<cstdio>
#define LL long long
using namespace std;
																																																				
const int N = 2e6 + 5 , Ml = -1e9 , Mr = 1e9;
const LL P = 1e9 + 7 , phi = 1e9 + 6;
int n , sz = 1;

struct segment{
	LL sum , tg1 , tg2;
	int ls , rs;
}seg[N];

LL fpow(LL x , LL y)
{
	LL res = 1;
	for(; y; y >>= 1)
	{
		if (y & 1) res = res * x % P;
		x = x * x % P;
	}
	return res;
}

void New(int k , int o)
{
	if (!o) 
	{
		if (!seg[k].ls) 
		seg[seg[k].ls = ++sz] = segment{1 , 1 , 1 , 0 , 0};
	}
	else 
	{
		if (!seg[k].rs)
		seg[seg[k].rs = ++sz] = segment{1 , 1 , 1 , 0 , 0}; 
	}
}

void pushup(int k)
{
	seg[k].sum = seg[seg[k].ls].sum * seg[seg[k].rs].sum % P;
}

void pushdown(int k , int l , int r)
{
	if (seg[k].tg2 != 1)
	{
		New(k , 0) , New(k , 1);
		seg[seg[k].ls].sum = fpow(seg[seg[k].ls].sum , seg[k].tg2);
		seg[seg[k].rs].sum = fpow(seg[seg[k].rs].sum , seg[k].tg2);
		seg[seg[k].ls].tg2 = seg[seg[k].ls].tg2 * seg[k].tg2 % phi;
		seg[seg[k].rs].tg2 = seg[seg[k].rs].tg2 * seg[k].tg2 % phi;
		seg[seg[k].ls].tg1 = fpow(seg[seg[k].ls].tg1 , seg[k].tg2);																																
		seg[seg[k].rs].tg1 = fpow(seg[seg[k].rs].tg1 , seg[k].tg2);
		seg[k].tg2 = 1;
	}	
	if (seg[k].tg1 != 1)
	{
		int mid = (l + r) >> 1;
		New(k , 0) , New(k , 1);
		seg[seg[k].ls].sum = seg[seg[k].ls].sum * fpow(seg[k].tg1 , mid - l + 1) % P;
		seg[seg[k].rs].sum = seg[seg[k].rs].sum * fpow(seg[k].tg1 , r - mid) % P;
		seg[seg[k].ls].tg1 = seg[seg[k].ls].tg1 * seg[k].tg1 % P;
		seg[seg[k].rs].tg1 = seg[seg[k].rs].tg1 * seg[k].tg1 % P;
		seg[k].tg1 = 1;
	}
}

void seg_mul(int l , int r , int k , int x , int y , int z)
{
	if (x <= l && r <= y)
	{
		seg[k].sum = seg[k].sum * fpow(z , r - l + 1) % P;
		seg[k].tg1 = seg[k].tg1 * z % P;
		return;
	}
	pushdown(k , l , r);
	int mid = (l + r) >> 1;
	if (x <= mid) New(k , 0) , seg_mul(l , mid , seg[k].ls , x , y , z);
	if (y > mid) New(k , 1) , seg_mul(mid + 1 , r , seg[k].rs , x , y , z);
	pushup(k);
}

void seg_pow(int l , int r , int k , int x , int y , int z)
{
	if (x <= l && r <= y)
	{
		seg[k].sum = fpow(seg[k].sum , z);
		seg[k].tg1 = fpow(seg[k].tg1 , z) , seg[k].tg2 = seg[k].tg2 * z % phi;
		return;
	}
	pushdown(k , l , r);
	int mid = (l + r) >> 1;
	if (x <= mid) New(k , 0) , seg_pow(l , mid , seg[k].ls , x , y , z);
	if (y > mid) New(k , 1) , seg_pow(mid + 1 , r , seg[k].rs , x , y , z);
	pushup(k);
}

LL seg_query(int l , int r , int k , int x , int y)
{
	if (x <= l && r <= y) return seg[k].sum;
	pushdown(k , l , r);
	int mid = (l + r) >> 1; LL res = 1;
	if (x <= mid && seg[k].ls) res = seg_query(l , mid , seg[k].ls , x , y);
	if (y > mid && seg[k].rs) res = res * seg_query(mid + 1 , r , seg[k].rs , x , y) % P;
	return res;
}

int main()
{
	freopen("segment.in" , "r" , stdin);
	freopen("segment.out" , "w" , stdout);
	scanf("%d" , &n);
	int op , l , r , w;
	seg[0] = seg[1] = segment{1 , 1 , 1 , 0 , 0};
	while (n--)
	{
		scanf("%d%d%d" , &op , &l , &r) , ++l;
		if (op == 0) scanf("%d" , &w) , seg_mul(Ml , Mr , 1 , l , r , w);
		else if (op == 1) scanf("%d" , &w) , seg_pow(Ml , Mr , 1 , l , r , w);
		else printf("%lld\n" , seg_query(Ml , Mr , 1 , l , r));
	}
}
上一篇:【洛谷P1502】窗口的星星


下一篇:旷视科技提出SPCNet:一种任意形状的场景文本检测算法