結果
問題 |
No.2210 equence Squence Seuence
|
ユーザー |
|
提出日時 | 2022-11-28 08:09:29 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,200 bytes |
コンパイル時間 | 318 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 119,680 KB |
最終ジャッジ日時 | 2024-10-05 04:22:52 |
合計ジャッジ時間 | 4,389 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 16 WA * 4 RE * 5 |
ソースコード
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 = [] for i in range(n - 1): 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]) 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))