[提高组集训2021] 古老的序列问题

一、题目

有一个长度为 \(n\) 的整数序列 \(a\),你需要回答 \(m\) 个询问,每次给出 \(L,R\),求下列式子的值:

\[\sum_{l=L}^R\sum_{r=l}^R(\max_{i=l}^r s_i)\cdot (\min_{i=l}^rs_i) \]

\(n,m\leq 10^5\)

二、解法

解法一

猫树分治的进阶版应用,我们把猫叔看作线段树,每次把询问在猫树上面下放,下放的方式:如果正好覆盖当前区间那么留在这个节点上,回溯时算答案;如果不经过中点那么直接下放;如果经过中点那么统计左边对右边的贡献。

主要问题是算左对右的贡献,我们拿一个指针在左半部分从后往前移动,记录到中点的最小值和最大值分别是 \(la,lb\),那么在右半部分可以分出四个段,含义分别是:最值都在左边取得、最大值在左边取得、最小值在左边取得、最值都在右边取得。

不难发现可以用双指针来找到这些段,然后对于右边的每个点维护左边的一个后缀对它的贡献,在移动左边指针的时候更新贡献即可,四个段对应的系数不同(如果在右边去最值那么要用右边的来乘,这个叫做系数),那么我们用四个线段树来维护系数即可,询问的时候是一个区间查询的过程。

那么时间复杂度 \(O(n\log^2n)\)要注意卡常啊

解法二

如果只有一次询问怎么办?移动右端点维护每个左端点的答案,把右端点贡献到询问上去即可。

如果有多个询问怎么办?移动右端点维护每个右端点对应每个左端点的答案,把历史和贡献到询问上去即可。

那么写两个单调栈\(+\)一个带区间乘法支持区间历史和查询的线段树即可,用矩阵乘法维护历史和,时间复杂度 \(O(n\log n)\)

三、总结

猫树进阶应用,一言以蔽之:线段树加 \(cdq\) 分治,注意利用过中点这个性质!

遇事不决,矩阵乘法,矩阵什么都可以维护!

//cat-tree divide
#include <cstdio>
#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;
const int M = 100005;
const int N = 400005;
const int MOD = 1e9+7;
int read()
{
	int x=0,f=1;char c;
	while((c=getchar())<'0' || c>'9') {if(c=='-') f=-1;}
	while(c>='0' && c<='9') {x=(x<<3)+(x<<1)+(c^48);c=getchar();}
	return x*f;
}
int n,m,s[M],z[M],a[M],b[M],ab[M],ans[M],f[N];
struct node
{
	int l,r,id;
	bool operator < (const node &b) const
	{
		return l>b.l;
	}
};vector<node> q[N];
struct seg
{
	int L,R,sum[N],sx[N],tag[N];
	int mul(int x,int y) {return 1ll*x*y%MOD;}
	void add(int i,int c)
	{
		sum[i]=(sum[i]+mul(sx[i],c))%MOD;
		tag[i]=(tag[i]+c)%MOD;
	}
	void build(int i,int l,int r,int *a)
	{
		if(i==1) L=l,R=r;
		sum[i]=tag[i]=0;
		if(l==r)
		{
			sx[i]=a[l];
			return ;
		}
		int mid=(l+r)>>1;
		build(i<<1,l,mid,a);
		build(i<<1|1,mid+1,r,a);
		sx[i]=(sx[i<<1]+sx[i<<1|1])%MOD;
	}
	void down(int i)
	{
		if(!tag[i]) return ;
		add(i<<1,tag[i]);
		add(i<<1|1,tag[i]);
		tag[i]=0; 
	}
	void ins(int i,int l,int r,int L,int R,int c)
	{
		if(L>r || l>R) return ;
		if(L<=l && r<=R)
		{
			add(i,c);
			return ;
		}
		int mid=(l+r)>>1;down(i);
		ins(i<<1,l,mid,L,R,c);
		ins(i<<1|1,mid+1,r,L,R,c);
		sum[i]=(sum[i<<1]+sum[i<<1|1])%MOD;
	}
	int ask(int i,int l,int r,int L,int R)
	{
		if(L>r || l>R) return 0;
		if(L<=l && r<=R) return sum[i];
		int mid=(l+r)>>1;down(i);
		return (ask(i<<1,l,mid,L,R)+
		ask(i<<1|1,mid+1,r,L,R))%MOD;
	}
	int Ask(int x) {return ask(1,L,R,L,x);}
}I,A,B,AB;
void add(int &x,int y) {x=(x+y)%MOD;}
void div(int i,int l,int r)
{
	if(l==r)
	{
		f[i]=1ll*s[l]*s[l]%MOD;
		for(auto x:q[i])
			add(ans[x.id],f[i]);
		return ;
	}
	int mid=(l+r)>>1;a[mid]=MOD;b[mid]=0;
	vector<node> v;
	for(auto x:q[i])
		if(x.l<=mid && x.r>mid && !(x.l==l && x.r==r))
			v.push_back(x);
	sort(v.begin(),v.end());
	for(int i=mid+1;i<=r;i++)
	{
		z[i]=1;
		a[i]=min(a[i-1],s[i]);
		b[i]=max(b[i-1],s[i]);
		ab[i]=1ll*a[i]*b[i]%MOD;
	}
	I.build(1,mid+1,r,z);A.build(1,mid+1,r,a);
	B.build(1,mid+1,r,b);AB.build(1,mid+1,r,ab);
	int pa=mid,pb=mid,la=MOD,lb=0,t=mid+1;
	for(int i=mid,j=0;i>=l;i--)
	{
		la=min(la,s[i]);lb=max(lb,s[i]);
		for(;pa<r && a[pa+1]>la;pa++);
		for(;pb<r && b[pb+1]<lb;pb++);
		I.ins(1,t,r,t,min(pa,pb),1ll*la*lb%MOD);
		if(pa<pb) A.ins(1,t,r,pa+1,pb,lb);
		if(pb<pa) B.ins(1,t,r,pb+1,pa,la);
		AB.ins(1,t,r,max(pa,pb)+1,r,1);
		while(j<v.size() && v[j].l>=i)
		{
			node o=v[j];
			add(ans[o.id],(1ll*I.Ask(o.r)+A.Ask(o.r)
			+B.Ask(o.r)+AB.Ask(o.r))%MOD);
			j++;
		}
	}
	f[i]=(1ll*I.Ask(r)+A.Ask(r)+B.Ask(r)+AB.Ask(r))%MOD;
	for(auto x:q[i])
	{
		if(l==x.l && x.r==r) continue;
		if(x.r<=mid) q[i<<1].push_back(x);
		else if(x.l>mid) q[i<<1|1].push_back(x);
		else
		{
			node t1=x,t2=x;
			t1.r=mid;q[i<<1].push_back(t1);
			t2.l=mid+1;q[i<<1|1].push_back(t2);
		}
	}
	div(i<<1,l,mid);div(i<<1|1,mid+1,r);
	f[i]=(1ll*f[i]+f[i<<1]+f[i<<1|1])%MOD;
	for(auto x:q[i]) if(l==x.l && x.r==r)
		add(ans[x.id],f[i]);
}
void write(int x)
{
	if(x<=9)
	{
		putchar(x+'0');
		return ;
	}
	write(x/10);
	putchar(x%10+'0');
}
signed main()
{
	freopen("sequence.in","r",stdin);
	freopen("sequence.out","w",stdout);
	n=read();m=read();
	for(int i=1;i<=n;i++)
		s[i]=read();
	for(int i=1;i<=m;i++)
	{
		int l=read(),r=read();
		q[1].push_back(node{l,r,i});
	}
	div(1,1,n);
	for(int i=1;i<=m;i++)
		write(ans[i]),puts("");
}
//segment tree maintaining matrix
#include <cstdio>
#include <vector>
#include <cstring>
#include <cassert>
using namespace std;
const int M = 100005;
const int MOD = 1e9+7;
int read()
{
	int x=0,f=1;char c;
	while((c=getchar())<'0' || c>'9') {if(c=='-') f=-1;}
	while(c>='0' && c<='9') {x=(x<<3)+(x<<1)+(c^48);c=getchar();}
	return x*f;
}
int n,m,ans[M],a[M],sa[M],sb[M];
struct node{int x,y;};vector<node> g[M];
struct mat
{
	int a[2][2];
	mat() {memset(a,0,sizeof a);}
	void reset() {a[0][1]=a[1][0]=0;a[0][0]=a[1][1]=1;}
	mat operator * (const mat &b) const
	{
		mat r;
		for(int i=0;i<2;i++)
			for(int j=0;j<2;j++)
				for(int k=0;k<2;k++)
					r.a[i][k]=(r.a[i][k]+
					1ll*a[i][j]*b.a[j][k])%MOD;
		return r;
	}
}tr[4*M],tag[4*M],ZXY;
int qkpow(int a,int b)
{
	int r=1;
	while(b>0)
	{
		if(b&1) r=1ll*r*a%MOD;
		a=1ll*a*a%MOD;
		b>>=1;
	}
	return r;
}
mat con(int c)
{
	mat r;
	r.a[0][0]=c;r.a[1][1]=1;
	return r;
}
void build(int i,int l,int r)
{
	tag[i].reset();
	tr[i].a[0][0]=r-l+1;
	if(l==r) return ;
	int mid=(l+r)>>1;
	build(i<<1,l,mid);
	build(i<<1|1,mid+1,r);
}
void down(int i)
{
	if(tag[i].a[0][0]==1 && tag[i].a[0][1]==0
	&& tag[i].a[1][0]==0 && tag[i].a[1][1]==1) return ;
	tr[i<<1]=tr[i<<1]*tag[i];
	tag[i<<1]=tag[i<<1]*tag[i];
	tr[i<<1|1]=tr[i<<1|1]*tag[i];
	tag[i<<1|1]=tag[i<<1|1]*tag[i];
	tag[i].reset();
}
void add(int i,int l,int r,int L,int R,mat c)
{
	if(L>r || l>R) return ;
	if(L<=l && r<=R)
	{
		tr[i]=tr[i]*c;
		tag[i]=tag[i]*c;
		return ;
	}
	int mid=(l+r)>>1;down(i);
	add(i<<1,l,mid,L,R,c);
	add(i<<1|1,mid+1,r,L,R,c);
	tr[i].a[0][0]=(tr[i<<1].a[0][0]
	+tr[i<<1|1].a[0][0])%MOD;
	tr[i].a[0][1]=(tr[i<<1].a[0][1]
	+tr[i<<1|1].a[0][1])%MOD;
}
int ask(int i,int l,int r,int L,int R)
{
	if(l>R || L>r) return 0;
	if(L<=l && r<=R) return tr[i].a[0][1];
	int mid=(l+r)>>1;down(i);
	return (ask(i<<1,l,mid,L,R)+
	ask(i<<1|1,mid+1,r,L,R))%MOD;
}
void write(int x)
{
	if(x<=9)
	{
		putchar(x+'0');
		return ;
	}
	write(x/10);
	putchar(x%10+'0');
}
signed main()
{
	freopen("sequence.in","r",stdin);
	freopen("sequence.out","w",stdout);
	n=read();m=read();
	for(int i=1;i<=n;i++)
		a[i]=read();
	build(1,1,n);
	for(int i=1;i<=m;i++)
	{
		int l=read(),r=read();
		g[r].push_back(node{l,i});
	}
	int la=0,lb=0;mat I;
	I.reset();I.a[0][1]=1;
	for(int i=1;i<=n;i++)
	{
		while(la && a[sa[la]]<a[i])
		{
			int c=qkpow(a[sa[la]],MOD-2);
			add(1,1,n,sa[la-1]+1,sa[la],con(c));
			la--;
		}
		while(lb && a[sb[lb]]>a[i])
		{
			int c=qkpow(a[sb[lb]],MOD-2);
			add(1,1,n,sb[lb-1]+1,sb[lb],con(c));
			lb--;
		}
		add(1,1,n,sa[la]+1,i,con(a[i]));
		add(1,1,n,sb[lb]+1,i,con(a[i]));
		sa[++la]=i;sb[++lb]=i;
		add(1,1,n,1,i,I);
		for(auto t:g[i])
			ans[t.y]=(ans[t.y]+ask(1,1,n,t.x,i))%MOD;
	}
	for(int i=1;i<=m;i++)
		write(ans[i]),puts("");
}
posted @ 2021-10-03 20:49  C202044zxy  阅读(483)  评论(0编辑  收藏  举报