题解 丑国传说 · 丑国旅游(En:Travel To The Ugly) By---阿丑

传送门是划水时口胡的一道大水题

前置知识:

可持久化线段树。

题意:

  • 给出一个长度为 \(n\) 的数列 \(a_i\)

  • \(m\) 次在线询问,每次询问求出 \(\sum\limits_{i=l}^{r-1}|x-a_i|\times|x-a_{i+1}|\)

  • \(n,m\le3\times10^5\)

为了简洁,设 \(f_{x,i}=|x-a_i|\times|x-a_{i+1}|\)

分析:

这题题面看起来就像线段树,但注意到对于每个 \(x\)\(f_{x,i}\) 都不相同,我们也开不下 \(10^9\) 棵线段树,所以暂时并不行。(对于 \(10\%\)\(x\le10\) 的数据,这一方法就非常可行)

不过离散化后,我们可能只需要开 \(3\times10^5\) 棵树,这提示我们往主席树的方向思考。

根据绝对值的定义,有:

\[|x-a|=\begin{cases}x-a&x\ge a\\-x+a &x<a\end{cases} \]

那么:

\[f_{x,i}=\begin{cases}x^2-(a_i+a_{i+1})x+a_ia_{i+1}&x\ge\max(a_i,a_{i+1})\text{ 或 }x<\min(a_i,a_{i+1})\\-x^2+(a_i+a_{i+1})x-a_ia_{i+1}&\min(a_i,a_{i+1})\le x<\max(a_i,a_{i+1})\end{cases} \]

只要 \(x\)\(a_i\)\(a_{i+1}\) 的关系确定后,我们就可以选择上述式子来求出答案,而不需要确定 \(x\) 的具体值。

具体地,假设 \(f_{x,i}=bx^2+cx+d\),则有 \(\sum f_{x,i}=x^2\sum b+x\sum c+\sum d\),可以用线段树维护 \(\sum b\)\(\sum c\)\(\sum d\),这样线段树就与 \(x\) 的具体值无关了。

注意到当 \(x\) 增大的过程中,\(x\)\(a_i\) 的关系最多改变一次(从 \(x\le a_i\)\(x>a_i\)),并且总是当 \(x\) 超过 \(a_i\) 的时候,\(f_{x,i}\)\(f_{x,i-1}\) 会改变。因此对于每个 \(a_i\) 建一棵线段树,并使用主席树维护这 \(n\) 棵树。每次 \(x\) 超过 \(a_i\) 时,在上一棵树的基础上修改 \(f_{x,i-1}\)\(f_{x,i}\),就可以解决问题了。

思路:

  1. 对每一个 \(a_i\) 开一棵线段树,并使用主席树维护。

  2. 对每个查询二分查找应该查询哪一棵线段树,直接查询即可。


code:

#include <bits/stdc++.h>
#define rep(i, l, r) for(register int i=l; i<=r; ++i)
#define rrep(i, r, l) for(register int i=r; i>=l; --i)
#define lfor(i, x) for(int i=head[x]; i; i=nxt[i])
#define ll long long
using namespace std;
inline ll read() {
	ll ret=0, k=1; char c; do if((c=getchar())=='-') k=-1; while(c<'0' || c>'9');
	while(c>='0' && c<='9') ret=(ret<<1)+(ret<<3)+(c^48), c=getchar(); return k*ret;
}
const int mN=3e5+100, mS=42*mN, mod=1e9+7;
int n, m, a[mN];
int ork, rk[mN];
int oe, head[mN], ver[mN], nxt[mN];
inline void add(int x, int y) {nxt[++oe]=head[x], ver[oe]=y, head[x]=oe;}

namespace Segment_Tree {
#define lc tree[p].son[0]
#define rc tree[p].son[1]
	int on, rt[mN];
	struct node {int son[2]; int b, c, d;} tree[mS];	//b c d 就是题解中提到的三个系数 
	inline void push_up(int p) {
		tree[p].b=(tree[lc].b+tree[rc].b)%mod;
		tree[p].c=(tree[lc].c+tree[rc].c)%mod;
		tree[p].d=(tree[lc].d+tree[rc].d)%mod;
	}
	void build(int &p, int l, int r) {
		p=++on;
		if(l==r) {
			if(1<=l && l<=n) tree[p].b=1, tree[p].c=(-a[l]-a[l+1])%mod, tree[p].d=(ll) a[l]*a[l+1]%mod;
			else tree[p].b=tree[p].c=tree[p].d=0;
		} else build(lc, l, l+r>>1), build(rc, (l+r>>1)+1, r), push_up(p);
	}
	void modify(int lp, int &p, int l, int r, int i) {	//修改 f(x,i) 的式子 
		if(p==lp || !p) p=++on, tree[p]=tree[lp];
		if(l==r) tree[p].b=-tree[p].b, tree[p].c=-tree[p].c, tree[p].d=-tree[p].d;	//取相反数即可 
		else {
			if(i<=(l+r>>1)) modify(tree[lp].son[0], lc, l, l+r>>1, i);
			else modify(tree[lp].son[1], rc, (l+r>>1)+1, r, i);
			push_up(p);
		}
	}
	int query(int p, int l, int r, int x, int y, int z) {
		if(x<=l && r<=y) return ((ll) tree[p].b*z%mod*z+(ll) tree[p].c*z+tree[p].d)%mod;	//即求 bz^2+cz+d 
		return ((lc&&x<=(l+r>>1)?query(lc, l, l+r>>1, x, y, z):0)
		       +(rc&&(l+r>>1)<y?query(rc, (l+r>>1)+1, r, x, y, z):0))%mod;
	}
#undef lc
#undef rc
}
using namespace Segment_Tree;

pair <int, int> tmp[mN];
void getrk() {	//去重,写得有点丑(
	rep(i, 1, n) tmp[i]=make_pair(a[i], i);
	sort(tmp+1, tmp+n+1);
	rep(i, 1, n) {
		if(tmp[i].first!=tmp[i-1].first) ++ork;
		rk[ork]=tmp[i].first, add(ork, tmp[i].second);
	}
}
int main() {
	n=read(), m=read();
	rep(i, 1, n) a[i]=read();
	getrk(), build(rt[0], 0, n);
	rep(i, 1, ork) lfor(t, i) modify(rt[i-1], rt[i], 0, n, ver[t]-1), modify(rt[i-1], rt[i], 0, n, ver[t]);
	//f(x, ver[t]) 和 f(x, ver[t]-1) 在 a[ver[t]] 变化时均需修改
	//因为修改时不想特判 ver[t]==1,所以下标从 0 到 n
	int ans=0, x, l, r;
	while(m--) {
		x=read()^ans, l=read()^ans, r=read()^ans;
		printf("%d\n", ans=(query(rt[upper_bound(rk, rk+ork+1, x)-rk-1], 0, n, l, r-1, x)+mod)%mod);
		//upper_bound-rk-1 找第一个小于 x 的数 
	}
	return 0;
}

本人代码常数巨大,欢迎各路神仙吊打。

本文来自博客园,作者:Maplisky,转载请注明原文链接:https://www.cnblogs.com/lbh2021/p/14945090.html

posted @ 2021-06-28 16:34  Maplisky  阅读(17)  评论(0编辑  收藏  举报