首页 > 技术文章 > BSOJ7526口胡

lmpp 2022-02-09 18:17 原文

直觉告诉我一般情况下,询问古怪的题都是分块,但是这一类题不太一样。

思考一个奇怪的暴力,每次询问的时候询问 \(f(1,k),f(2,k+1),f(3,k+2),...f(n-k+1,n)\),然后加起来一定是答案。

差分,思考 \(f(l+1,r+1)-f(l,r)\) 是多少。容易知道其对答案的贡献为 \((n-r)\)

考虑 \(l\)\(r+1\) 两个位置。

接下来设 \(pre[i]\) 为上一个颜色与自身相同的最靠右的位置,\(nxt[i]\) 类似。

可以发现 \(f(l+1,r+1)-f(l,r)=[pre[r+1]<l]-[nxt[l]>r+1]\)

问题转化为 \(\sum_{i=1}^{n-k}([pre[i+k]<i]-[nxt[i]>i+k])(n+1-k-i)\)

最后只需要加上 \([1,k]\) 的颜色个数乘上 \(n-k+1\) 即可。

推一推上面的东西:

\[\sum_{i=1}^{n-k}([k<(i+k-pre[i+k])]-[nxt[i]-i>k])(n+1-k-i) \]

这是经典的二维偏序,因为带修所以可以随随便便做到 \(O(n\log^2n)\)。但是这太简单了!所以其实有 \(O(n\log n)\) 的做法。

注意到后面的权值随随便便维护,所以我们只需要维护这个范围即可。

\(x[i]=i-pre[i],y[i]=nxt[i]-i\),那么我们询问的实际上是 \((\sum_{i=k+1}^n[k<x[i]](n-i+1))+(\sum_{i=1}^{n-k}[k<y[i]](n-k-i+1))\)

注意到很明显在 \(i\leq k\) 时有 \(x[i]\leq k\),且 \(n-k+1\leq i\) 时也有 \(y[i]\leq k\),所以实际上这两部分都不会被算入贡献,直接开两颗线段树即可。

至于如何 \(O(n\log n)\) 询问前缀颜色个数,用 \(pre\)\(nxt\) 随便维护一下就好了,具体可以参考 BSOJ7791。

推荐阅读