結果
問題 |
No.2759 Take Pictures, Elements?
|
ユーザー |
![]() |
提出日時 | 2024-05-18 13:07:05 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 794 ms / 2,000 ms |
コード長 | 768 bytes |
コンパイル時間 | 187 ms |
コンパイル使用メモリ | 82,652 KB |
実行使用メモリ | 84,984 KB |
最終ジャッジ日時 | 2025-06-20 13:19:54 |
合計ジャッジ時間 | 3,305 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 21 |
ソースコード
# 2次元dp # TLEしたのでstdin、それでもTLEしたのでdpの更新の順番で枝刈り import sys input = sys.stdin.readline from collections import defaultdict N, Q = map(int, input().split()) A = list(map(int, input().split())) B = list(map(int, input().split())) pos = defaultdict(list) for i in range(N): pos[A[i]].append(i) #print(pos) INF = 10**10 dp = [[INF]*N for i in range(Q+1)] dp[0][0] = 0 #print(dp[0]) for i in range(1, Q+1): num = B[i-1] for j in range(N): if dp[i-1][j] == INF: continue for k in pos[num]: dp[i][k] = min(dp[i][k], dp[i-1][j]+abs(j-k)) #print('num', num, 'pos[num]', pos[num], dp[i]) ans = INF for k in pos[B[-1]]: ans = min(ans, dp[Q][k]) print(ans)