結果

問題 No.2114 01 Matching
ユーザー first_vil
提出日時 2022-10-01 22:11:32
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,286 bytes
コンパイル時間 170 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 150,292 KB
最終ジャッジ日時 2024-06-29 09:59:24
合計ジャッジ時間 9,142 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other TLE * 1 -- * 50
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict, deque
import sys

int1 = lambda x: int(x) - 1

# input = lambda: sys.stdin.buffer.readline()
input = lambda: sys.stdin.readline().rstrip()
ii = lambda: int(input())
i1 = lambda: int1(input())
mi = lambda: map(int, input().split())
mi1 = lambda: map(int1, input().split())
li = lambda: list(mi())
li1 = lambda: list(mi1())
lli = lambda n: [li() for _ in range(n)]

INF = float("inf")
# mod = int(1e9 + 7)
mod = 998244353

n, m, k = mi()
b = li()
r = li()

db = defaultdict(list)
dr = defaultdict(list)
for bb in b:
    db[bb % k].append(bb)
for rr in r:
    dr[rr % k].append(rr)

cost = 0
cnt = 0


def solve(x, y):
    n = len(x)
    m = len(y)
    dp = [INF] * (m + 1)
    dp[0] = 0
    for i in range(n):
        nxt = [INF] * (m + 1)
        mn = INF
        for j in range(m):
            if mn > dp[j]:
                mn = dp[j]
            if nxt[j + 1] > abs(x[i] - y[j]) + mn:
                nxt[j + 1] = abs(x[i] - y[j]) + mn
        dp = nxt
    return dp[m]

for ky in db.keys():
    if ky not in dr:
        continue
    x = deque(sorted(db[ky][:]))
    y = deque(sorted(dr[ky][:]))

    if len(x) > len(y):
        x, y = y, x

    cost += solve(x, y)
    cnt += len(x)

if cnt == min(n, m):
    print(cost // k)
else:
    print(-1)
0