有一个长度为n的字符串,每一位只会是p或j。你需要取出一个子串S(从左到右或从右到左一个一个取出),使得不管是从左往右还是从右往左取,都保证每时每刻已取出的p的个数不小于j的个数。你需要最大化|S|。
这道题先推一下式子。满足从左到右条件时需要sum[p][i]-sum[p][l-1]>=sum[j][i]-sum[j][l-1],那移一下,变成sum[p][i]-sum[j][i]>=sum[p][l-1]-sum[j][l-1]。从右到左同理,所以我们就把一个字符串的每一位变成sum[p][i]-sum[j][i],题目条件就变成左端点是这个子串的最小值,右端点是这个子串的最大值。
那解决这个用一个单调栈,再加一个rmq就可以了。
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdlib>
using namespace std;
inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0' || ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0' && ch<='9')x=x*10+ch-'0',ch=getchar();
return x*f;
}
inline void write(int x)
{
if(x<0)putchar('-'),x=-x;
if(x>9)write(x/10);
putchar(x%10+'0');
}
int a[1000010],sta[1000010],fa[1000010],sat[1000010];
char ss[1000010];
inline int findfa(int x)
{
if(fa[x]!=x)fa[x]=findfa(fa[x]);
return fa[x];
}
int main()
{
//freopen("3521.in","r",stdin);
//freopen("3521.out","w",stdout);
int n=read(),s0=0,s1=0;
scanf("%s",ss+1);
for(int i=1;i<=n;i++)
{
if(ss[i]=='p')s0++;
else s1++;
a[i]=s0-s1;
}
int top=0,ans=0,now=0;
for(int i=0;i<=n;i++)
{
fa[i]=i;
while(now && a[i]<a[sat[now]])fa[sat[now]]=i,now--;
sat[++now]=i;
while(top && a[sta[top]]<=a[i])top--;
int p=sta[top]+1;if(top==0)p=0;
ans=max(ans,i-findfa(p));
sta[++top]=i;
}
printf("%d\n",ans);
return 0;
}