#436. 子串的最大差(单调栈)
Posted MangataTS
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了#436. 子串的最大差(单调栈)相关的知识,希望对你有一定的参考价值。
题目链接
http://oj.daimayuan.top/problem/436
题面
思路
我们考虑每一个点作为一个区间最小值和区间最大值的次数,那么我们可以从两边延申,对于区间最小值而言找到左边第一个大于自身的数,对于右边也找到大于第一个大于自身的数,那么这两个区间长度的积就是我们区间以当前为最小值的次数,很显然这个值可以通过单调栈维护,那么我们分别从左往右和从右往左用两个单调栈维护,注意只有一个地方取=
即可,同理可以将每个点作为区间最大值的次数统计起来,统计完信息后我们从左往右将每个点的权值计算在ans
里即可,即:
a
n
s
+
=
a
[
i
]
∗
(
l
m
a
x
[
i
]
∗
r
m
a
x
[
i
]
−
l
m
i
n
[
i
]
∗
r
m
i
n
[
i
]
)
ans += a[i] * (lmax[i] * rmax[i] - lmin[i] * rmin[i])
ans+=a[i]∗(lmax[i]∗rmax[i]−lmin[i]∗rmin[i])
代码
#include<bits/stdc++.h>
using namespace std;
//----------------自定义部分----------------
#define ll long long
#define mod 1000000007
#define endl "\\n"
#define PII pair<int,int>
#define INF 0x3f3f3f3f
int dx[4] = -1, 0, 1, 0, dy[4] = 0, 1, 0, -1;
ll ksm(ll a,ll b)
ll ans = 1;
for(;b;b>>=1LL)
if(b & 1) ans = ans * a % mod;
a = a * a % mod;
return ans;
ll lowbit(ll x)return -x & x;
const int N = 2e6+10;
//----------------自定义部分----------------
ll t,n,m,q,a[N],b[N],lmin[N],rmin[N],lmax[N],rmax[N];
void slove()
cin>>n;
for(int i = 1;i <= n; ++i) cin>>a[i],b[i] = a[i];
reverse(b+1,b+n+1);
stack<int> S;
//找每个数作为min被调用多少次
//左半边
for(int i = 1;i <= n; ++i)
while(S.size() && a[S.top()] >= a[i]) S.pop();
if(S.size()) lmin[i] = i - S.top();
else lmin[i] = i;
S.push(i);
//右半边
while(S.size()) S.pop();
for(int i = n;i >= 1; --i)
while(S.size() && a[S.top()] > a[i]) S.pop();
if(S.size()) rmin[i] = S.top() - i;
else rmin[i] = n - i + 1;
S.push(i);
//找每个数作为max被调用多少次
while(S.size()) S.pop();
for(int i = 1;i <= n; ++i)
while(S.size() && a[S.top()] <= a[i]) S.pop();
if(S.size()) lmax[i] = i - S.top();
else lmax[i] = i;
S.push(i);
while(S.size()) S.pop();
for(int i = n;i >= 1; --i)
while(S.size() && a[S.top()] < a[i]) S.pop();
if(S.size()) rmax[i] = S.top() - i;
else rmax[i] = n - i + 1;
S.push(i);
ll ans = 0LL;
for(int i = 1;i <= n; ++i)
ans += a[i] * (lmax[i] * rmax[i] - lmin[i] * rmin[i]);
cout<<ans<<endl;
int main()
ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
t = 1;
while(t--)
slove();
return 0;
以上是关于#436. 子串的最大差(单调栈)的主要内容,如果未能解决你的问题,请参考以下文章
POJ 3415 Common Substrings(长度不小于K的公共子串的个数+后缀数组+height数组分组思想+单调栈)