結果

問題 No.2210 equence Squence Seuence
ユーザー ShirotsumeShirotsume
提出日時 2022-11-28 08:04:30
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 178 ms / 2,000 ms
コード長 1,226 bytes
コンパイル時間 226 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 130,688 KB
最終ジャッジ日時 2024-11-18 00:56:06
合計ジャッジ時間 5,701 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
from collections import deque, Counter
sys.setrecursionlimit(5 * 10 ** 5)
from pypyjit import set_param
set_param('max_unroll_recursion=-1')
input = lambda: sys.stdin.readline().rstrip()
ii = lambda: int(input())
mi = lambda: map(int, input().split())
li = lambda: list(mi())
inf = 2 ** 63 - 1
mod = 998244353

def solve(n, k, p):
    top = []
    bottom = []
    same = []
    p.append(0)

    for i in range(n):
        if p[i] < p[i + 1]:
            top.append(i)
            for v in same:
                top.append(v)
            same = []
        elif p[i] > p[i + 1]:
            bottom.append(i)
            for v in same:
                bottom.append(v)
            same = []
        else:
            same.append(i)
    q = top + bottom[::-1]
    ind = q[n - k]
    ans = []
    for i in range(n):
        if i == ind:
            continue
        ans.append(p[i])
    p.pop()
    return ans

def solveg(n, k, a):
    X = []
    for i in range(n):
        x = []
        for j in range(n):
            if i == j:
                continue
            x.append(a[j])
        X.append(x)

    X.sort()
    return X[k - 1]

n, k = mi()

a = li()

print(*(solve(n, k, a) if n > 7000 else solveg(n, k, a)))
0