[CF997E] Good Subsegments
前言
我后悔了,要是拉这个加强版就不至于人均切 T4 了。
题目
洛谷
CF
讲解
CF526F Pudding Monsters 的加强版,没做过的建议先去做一下,至少要熟悉那道题的思路。
我们还是移动右端点,用单调栈+线段树维护一下 \(max-min+cnt\) 的值,思考一个好区间对之后的区间的贡献是什么?
其实就是1啦,只要后面的区间的左端点达到现在这个好区间的左端点,那么就会有贡献,所以我们考虑维护一个标记,将其放在好区间的左端点,一旦后面的区间包含这个左端点,那么就可以获得它的贡献。
那么我们应该怎么快速给这些好区间的左端点加上一个1的标记呢?它们的通性不就是全局最小值吗?所以我们只需要把所以是最小值的地方都加上这个标记即可。由于之前的单调栈+线段树的区间加操作不会影响一个顶点左右儿子最小值相对关系,所以只要儿子和当前节点的最小值相同,我们就可以放心大胆地下传这个标记。
时间复杂度 \(O(n\log_2n)\)。
代码
easy to understand
//12252024832524
#include
#define TT template
using namespace std;
typedef long long LL;
const int MAXN = 120005;
int n,m;
int a[MAXN];
LL ans[MAXN];
vector > q[MAXN];
LL Read()
{
LL x = 0,f = 1; char c = getchar();
while(c > '9' || c < '0'){if(c == '-') f = -1;c = getchar();}
while(c >= '0' && c <= '9'){x = (x*10) + (c^48);c = getchar();}
return x * f;
}
TT void Put1(T x)
{
if(x > 9) Put1(x/10);
putchar(x%10^48);
}
TT void Put(T x,char c = -1)
{
if(x < 0) putchar('-'),x = -x;
Put1(x); if(c >= 0) putchar(c);
}
TT T Max(T x,T y){return x > y ? x : y;}
TT T Min(T x,T y){return x < y ? x : y;}
TT T Abs(T x){return x < 0 ? -x : x;}
#define lc (x<<1)
#define rc (x<<1|1)
int MIN[MAXN<<2],lz[MAXN<<2],cnt[MAXN<<2],tim[MAXN<<2];
LL val[MAXN<<2];
void calc(int x,LL v)
{
val[x] += v * cnt[x];
tim[x] += v;
}
void down(int x)
{
if(lz[x])
{
MIN[lc] += lz[x]; lz[lc] += lz[x];
MIN[rc] += lz[x]; lz[rc] += lz[x];
lz[x] = 0;
}
if(tim[x])//这两个if写反了,调了5分钟
{
if(MIN[x] == MIN[lc]) calc(lc,tim[x]);
if(MIN[x] == MIN[rc]) calc(rc,tim[x]);
tim[x] = 0;
}
}
void up(int x)
{
MIN[x] = Min(MIN[lc],MIN[rc]); cnt[x] = 0;
if(MIN[x] == MIN[lc]) cnt[x] += cnt[lc];
if(MIN[x] == MIN[rc]) cnt[x] += cnt[rc];
}
void Build(int x,int l,int r)
{
cnt[x] = r-l+1;
if(l == r) return;
int mid = (l+r) >> 1;
Build(lc,l,mid); Build(rc,mid+1,r);
}
void Add(int x,int l,int r,int ql,int qr,int v)
{
if(ql <= l && r <= qr)
{
MIN[x] += v;
lz[x] += v;
return;
}
int mid = (l+r) >> 1;
down(x);
if(ql <= mid) Add(lc,l,mid,ql,qr,v);
if(mid+1 <= qr) Add(rc,mid+1,r,ql,qr,v);
up(x);
}
LL Query(int x,int l,int r,int ql,int qr)
{
if(ql <= l && r <= qr) return val[x];
int mid = (l+r) >> 1;LL ret = 0;
down(x);
if(ql <= mid) ret += Query(lc,l,mid,ql,qr);
if(mid+1 <= qr) ret += Query(rc,mid+1,r,ql,qr);
return ret;
}
int s1[MAXN],tl1,s2[MAXN],tl2;
int main()
{
// freopen(".in","r",stdin);
// freopen(".out","w",stdout);
n = Read();
for(int i = 1;i <= n;++ i) a[i] = Read();
m = Read();
for(int i = 1;i <= m;++ i)
{
int l = Read();
q[Read()].emplace_back(make_pair(l,i));
}
Build(1,1,n);
for(int i = 1;i <= n;++ i)
{
Add(1,1,n,1,i,-1);
while(tl1 && a[i] > a[s1[tl1]]) Add(1,1,n,s1[tl1-1]+1,s1[tl1],-a[s1[tl1]]),--tl1;
while(tl2 && a[i] < a[s2[tl2]]) Add(1,1,n,s2[tl2-1]+1,s2[tl2],a[s2[tl2]]),--tl2;
s1[++tl1] = s2[++tl2] = i;
Add(1,1,n,s1[tl1-1]+1,s1[tl1],a[s1[tl1]]);
Add(1,1,n,s2[tl2-1]+1,s2[tl2],-a[s2[tl2]]);
calc(1,1);
for(auto A : q[i]) ans[A.second] = Query(1,1,n,A.first,i);
}
for(int i = 1;i <= m;++ i) Put(ans[i],'\n');
return 0;
}