結果

問題 No.3075 Mex Recurrence Formula
コンテスト
ユーザー PNJ
提出日時 2025-03-28 22:16:58
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 104 ms / 2,000 ms
コード長 643 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 225 ms
コンパイル使用メモリ 95,984 KB
実行使用メモリ 121,020 KB
最終ジャッジ日時 2026-07-08 01:22:10
合計ジャッジ時間 6,454 ms
ジャッジサーバーID
(参考情報)
judge3_0 / judge2_1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

N, X = map(int, input().split())
A = list(map(int, input().split()))
C = [0 for _ in range(N + 1)]
for a in A:
  if a <= N:
    C[a] += 1
for a in range(N + 1):
  if C[a] == 0:
    A.append(a)
    break
D = [-1 for _ in range(N + 1)]
for i in range(N + 1):
  if A[i] <= N:
    D[A[i]] = i

i = 0
R = [-1 for _ in range(N + 1)]
for a in range(N + 1):
  j = D[a]
  if j != -1:
    if R[j] >= 0:
      j = i
      while R[j] >= 0:
        j += 1
      R[j] = a
      j += 1
      i = j
    else:
      R[j] = a
  else:
    while R[i] >= 0:
      i += 1
    R[i] = a
    i += 1
if X <= N + 1:
  print(A[X - 1])
else:
  print(R[(X - 1) % (N + 1)])
0