題目
話說世界上有很多超級英雄:蝙蝠俠,蜘蛛俠,超人,名字都寫不出來的人等等。在他們之中有一個叫Kickass。今天他想模仿蜘蛛俠,所以他選擇了一排高樓來跳。
具體來說,他選擇了一列N幢高樓,從左到右標號爲1到N。一開始他在第K幢高樓。不幸的是,Kickass能力非常有限,只能跳到向左或向右相鄰的高樓,而且他要跳到的樓的高度必須不能大於他現在處在的樓的高度。Kickass不想自己看起來很渣渣,所以他在一些高樓頂部放了蹦牀,從這些高樓起跳,能跳到任何其他的高樓,不管要跳到的高樓在哪裏或是多高。
你的任務是找到Kickass在第K高樓起跳能跳到的最多不同的高樓數。如果Kickass跳到一幢高樓超過一次,我們只會算一次。並且,即使Kickass沒有重新跳到,第K幢高樓還是要算入答案。
這個比較明顯可以用暴力做:
我們可以發現所有彈簧分別向左、向右連續單調不減的樓都可以到達,那麼先打個標記,並統計入答案內(若s不能到達彈簧,則暴力求s向左、向右最長的單調不增的樓加上等高反向的一段),然後再找沒被標記的最長單調不減或不增的一段,加入答案,輸出ans就行了
這些都可以在O(n)做出來
貼代碼
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
#define N 300011
int n,s,ans;
int a[N];
bool bz[N];
bool p;
void init(){
scanf("%d %d",&n,&s);
for (int i=1;i<=n;i++)
scanf("%d",&a[i]);
}
void pre(){
static char c;
static int x;
for (int i=1;i<=n;i++){
scanf(" %c",&c);
if (c=='T'){
p=1;
x=i;
bz[x]=1;
for (--x;x&&a[x]>=a[x+1]&&!bz[x];bz[x--]=1);
x=i;
for (++x;x<=n&&a[x]>=a[x-1]&&!bz[x];bz[x++]=1);
}
}
if (!bz[s])p=0;
if (!p)return;
}
void work(){
static int x,y,z;
if (p){
bz[0]=1,bz[n+1]=1;
y=0,z=0;
for (int i=1;i<=n;i++)
if (bz[i]){
ans++;
z=max(y,z);
y=0;
}else
if (bz[i-1])
y=1;
else
if (a[i-1]>=a[i])
++y;
else{
z=max(y,z);
y=1;
}
z=max(y,z);
y=0;
for (int i=n;i;i--)
if (bz[i]){
z=max(y,z);
y=0;
}else
if (bz[i+1])
y=1;
else
if (a[i+1]>=a[i])
++y;
else{
z=max(y,z);
y=1;
}
z=max(y,z);
ans+=z;
}else{
x=s;
ans=1;
for (--x;x&&a[x]<=a[x+1];ans++,x--);
x=s;
for (;x<n&&a[x+1]==a[x];++x,ans++);
x=s;
z=1;
for (x++;x<=n&&a[x]<=a[x-1];z++,x++);
x=s;
for(;x>1&&a[x-1]==a[x];--x,z++);
ans=max(ans,z);
}
}
void write(){
printf("%d",ans);
}
int main(){
init();
pre();
work();
write();
return 0;
}