Description
话说世界上有很多超级英雄:蝙蝠侠,蜘蛛侠,超人,名字都写不出来的人等等。在他们之中有一个叫Kickass。今天他想模仿蜘蛛侠,所以他选择了一排高楼来跳。
具体来说,他选择了一列N幢高楼,从左到右标号为1到N。一开始他在第K幢高楼。不幸的是,Kickass能力非常有限,只能跳到向左或向右相邻的高楼,而且他要跳到的楼的高度必须不能大于他现在处在的楼的高度。Kickass不想自己看起来很渣渣,所以他在一些高楼顶部放了蹦床,从这些高楼起跳,能跳到任何其他的高楼,不管要跳到的高楼在哪里或是多高。
你的任务是找到Kickass在第K高楼起跳能跳到的最多不同的高楼数。如果Kickass跳到一幢高楼超过一次,我们只会算一次。并且,即使Kickass没有重新跳到,第K幢高楼还是要算入答案。
Input
第一行输入两个数:高楼数 N(3<=N<=300,000),起跳位置K(1<=K<=N)。
第二行包括N个小于10^6的整数,表示1到N高楼的高度。
第三行包括一行N个字符’.’或’T’。如果第i个字符为’T’,则第i幢楼上有蹦床。
Output
输出一行,最多可以跳到的高楼数。
Sample Input
输入1:
6 4
12 16 16 16 14 14
.T…
输入2:
10 1
10 7 3 1 1 9 8 2 4 10
…T…T…
Sample Output
输出1:
5
输出2:
7
Data Constraint
3<=N<=300,000
Hint
第二个样例Kickass经过高楼的顺序:
1 à 2 à 3 à 6 à 10 à 9 à 8.
Solution
这题我们可以找一找规律。
我们发现:
如果能够走到一个蹦床的话,我们就可以跳到任何一个蹦床
而对于一个蹦床,我们可以发现,它的左边不递减以及右边不递增都是可以走到的:
3 2 1 2 3
…T…
就是3跳到5再跑回3,再跳到1再跑回3。
我们只要分情况(能到蹦床以及不能到蹦床)分开求即可。
还有,到了蹦床的话,我们发现还有一次机会可以跳到一个点。
所以,我们再求一下剩下的最长连续不递减或不递增的序列即可。
Code
#include<cstdio>
#include<algorithm>
#define N 300010
using namespace std;
int n,K,a[N],b[N],le,ri,tot=0,j,ma=0,maxx=0;
int bz[N],ans=0,s=0,s1=0,s2=0;
bool check[N];
char c;
inline int read()
{
int x=0,f=0; char c=getchar();
while (c<'0' || c>'9') f=(c=='-') ? 1:f,c=getchar();
while (c>='0' && c<='9') x=(x<<1)+(x<<3)+(c^48),c=getchar();
return f ? -x:x;
}
int gsum(int l,int r)
{
int s=0;
for (int i=l;i<=r;i++) s+=(!bz[i]);
return s;
}
int gans(int l,int r)
{
int s=0;
for (int i=l;i<=r;i++) s+=(bz[i]);
return s;
}
void color(int l,int r) {for (int i=l;i<=r;i++) bz[i]=1;}
int last()
{
int zuo=0,s=0;
for (int i=1;i<=n;i++)
{
if (bz[i])
{
if (zuo) s=max(s,i-zuo),zuo=0;
continue;
}
if (!zuo) zuo=i;
else if (a[i]<a[i-1])
{
s=max(s,i-zuo),zuo=i;
continue;
}
s=max(s,i-zuo+1);
}
for (int i=1;i<=n;i++)
{
if (bz[i])
{
if (zuo) s=max(s,i-zuo),zuo=0;
continue;
}
if (!zuo) zuo=i;
else if (a[i]>a[i-1])
{
s=max(s,i-zuo),zuo=i;
continue;
}
s=max(s,i-zuo+1);
}
return s;
}
int main()
{
freopen("man.in","r",stdin);
// freopen("man.out","w",stdout);
n=read(),K=read();bz[K]=1;
for (int i=1;i<=n;i++) a[i]=read();
scanf("
");
for (int i=1;i<=n;i++)
if ((c=getchar())=='T')
{
bz[i]=check[i]=1;
j=i; while (a[j]<=a[j-1] && j>1) j--,bz[j]=1;
j=i; while (a[j]<=a[j+1] && j<n) j++,bz[j]=1;
}
ri=K; while (a[ri]==a[ri+1] && ri<n) ri++,bz[ri]=1,s+=check[ri];
le=K; while (a[le]==a[le-1] && le>1) le--,bz[le]=1,s+=check[le];
j=le; while (a[j]>=a[j-1] && j>1) j--,s1=check[j] ? j:s1;
maxx=ri-j+1;
if (s1) ma=gsum(s1,le-1);
j=ri; while (a[j]>=a[j+1] && j<n) j++,s2=check[j] ? j:s2;
maxx=max(maxx,j-le+1);
if (s2 && gsum(ri+1,j)>ma) color(ri+1,s2);
else if (ma) color(s1,le-1);
if (s1 || s2 || s) printf("%d
",gans(1,n)+last());
else printf("%d
",maxx);
return 0;
}