結果

問題 No.2210 equence Squence Seuence
コンテスト
ユーザー FromBooska
提出日時 2023-03-08 11:08:49
言語 PyPy3
(7.3.23 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 132 ms / 2,000 ms
+ 832µs
コード長 1,566 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 239 ms
コンパイル使用メモリ 96,104 KB
実行使用メモリ 136,448 KB
最終ジャッジ日時 2026-08-29 23:23:57
合計ジャッジ時間 6,018 ms
ジャッジサーバーID
(参考情報)
judge3_0 / judge1_0
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

# 前の数字から見ていく
# その番号を選んだら順位が何番から何番に入るのか
# その番号を選ばないときに落ちる番号(first or last)がKの場合はその番号を選ばない=skip
# 嫌なのが同じ番号が続くとき、その場合は最終的に次の番号が上か下か調べる必要ある
# ここが二重ループになるので高速化した

N, K = map(int, input().split())
A = list(map(int, input().split()))

# 高速化のために次が上がっているのか(+1)、下がっているのか(-1)を記録
updown = [0]*(N)
for i in range(N-1, -1, -1):
    if i == N-1:
        updown[i] = 1 #最後にINFがあるとみなす
    else:
        if A[i] < A[i+1]:
            updown[i] = 1
        elif A[i] > A[i+1]:
            updown[i] = -1
        elif A[i] == A[i+1]:
            updown[i] = updown[i+1]

last = N
first = 1
skip = -1
for i in range(0, N-1):
    if skip > 0:
        break

    if A[i] < A[i+1]:
        if K == last:
            skip = i
            break
        last -= 1
    elif A[i] > A[i+1]:
        if K == first:
            skip = i
            break
        first += 1
    elif A[i] == A[i+1]:
        if updown[i] == -1:
            if K == first:
                skip = i
                break
            first += 1
        elif updown[i] == 1:
            if K == last:
                skip = i
                break
            last -= 1            
    
if skip == -1:
    skip = N-1

ans = []
for i in range(N):
    if i != skip:
        ans.append(A[i])
print(*ans)
0