結果

問題 No.3075 Mex Recurrence Formula
ユーザー PNJ
提出日時 2025-03-28 22:16:58
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 128 ms / 2,000 ms
コード長 643 bytes
コンパイル時間 289 ms
コンパイル使用メモリ 82,536 KB
実行使用メモリ 111,568 KB
最終ジャッジ日時 2025-03-28 22:17:05
合計ジャッジ時間 6,972 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff # 

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