Submission #65875497

Source Code Expand

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
n=int(input())
a=list(map(int,input().split()))
cnt=[0]*(n-1)
for i in range(n-1):
    if a[i]<a[i+1]:
        cnt[i]=1
    elif a[i]>a[i+1]:
        cnt[i]-=1
p=[0]*n
v=[0]*n
for i in range(1,n-1):
    if cnt[i-1]==1 and cnt[i]==-1:
        p[i]=1
    elif cnt[i-1]==-1 and cnt[i]==1:
        v[i]=1
inf=10**9
np=[inf]*(n+1)
nv=[inf]*(n+1)
for i in range(n-1,-1,-1):
    np[i]=i if (i<n and v[i])else np[i+1]
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX

n=int(input())
a=list(map(int,input().split()))

cnt=[0]*(n-1)
for i in range(n-1):
    if a[i]<a[i+1]:
        cnt[i]=1
    elif a[i]>a[i+1]:
        cnt[i]-=1
p=[0]*n
v=[0]*n
for i in range(1,n-1):
    if cnt[i-1]==1 and cnt[i]==-1:
        p[i]=1
    elif cnt[i-1]==-1 and cnt[i]==1:
        v[i]=1
inf=10**9
np=[inf]*(n+1)
nv=[inf]*(n+1)
for i in range(n-1,-1,-1):
    np[i]=i if (i<n and v[i])else np[i+1]
    nv[i]=i if (i<n and v[i])else nv[i+1]
l=0
r=0
pc=0
vc=0
ans=0
for l in range(n):
    if r<=l:r=l+1
    while r<n and pc+p[r-1]<=1 and vc+v[r-1]<=1:
        pc+=p[r-1]
        vc+=v[r-1]
        r+=1
    if l+1<n and cnt[l]==1:
        pp=np[l+1]
        vv=nv[l+1]
        if pp<inf and vv<inf:
            rr=max(pp,vv)+1
            if rr<=r:
                ans+=r-rr
    if l+1<r:
        pc-=p[l+1]
        vc-=v[l+1]
print(ans)

Submission Info

Judge Result