結果

問題 No.2759 Take Pictures, Elements?
ユーザー Cecil
提出日時 2024-05-17 22:25:29
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 639 bytes
コンパイル時間 199 ms
コンパイル使用メモリ 82,384 KB
実行使用メモリ 65,352 KB
最終ジャッジ日時 2025-06-20 13:18:21
合計ジャッジ時間 1,952 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 11 WA * 10
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import defaultdict as dd
import bisect

N,Q = map(int, input().split())
A = list(map(int, input().split()))
B = list(map(int, input().split()))
idx = dd(list)
for i in range(N):
    idx[A[i]].append(i)
ans = 0
pre = 0
for i in range(Q):
    #print("###")
    #print(pre)
    r = bisect.bisect_left(idx[B[i]], pre)
    r = min(len(idx[B[i]])-1,r)
    l = max(0, r-1)
    #print(idx[B[i]][r],idx[B[i]][l])
    #print("###")
    if abs(idx[B[i]][r]-pre)<abs(idx[B[i]][l]-pre):
        ans += abs(idx[B[i]][r]-pre)
        pre = idx[B[i]][r]
    else:
        ans += abs(idx[B[i]][l]-pre)
        pre = idx[B[i]][l]
print(ans)
0