結果

問題 No.2287 ++ -- *=a /=a
ユーザー sapphire__15
提出日時 2023-04-23 11:00:36
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,348 ms / 2,000 ms
コード長 1,491 bytes
コンパイル時間 379 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 13,056 KB
最終ジャッジ日時 2024-11-17 20:03:50
合計ジャッジ時間 8,443 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

from typing import List, Tuple
import sys

def solve(x: int, y: int, a: int) -> int:
    fromX: List[Tuple[int, int]] = []
    c: int = 0
    while 0 < x:
        fromX.append((x, c))
        x //= a
        c += 1
    fromX.append((x, c))

    dp: List[Tuple[int, int, int]] = [(y, 0, 1e9)]
    while 1:
        [i, j, k] = dp[-1]
        d, r = i // a, i % a
        if r == 0:
            dp.append((d,
                       min(j + 1, k + 2),
                       k + a))
        elif r == a - 1:
            dp.append((d,
                       j + a,
                       min(j + 2, k + 1)))
        else:
            dp.append((d,
                      min(j + r + 1, k + r + 2),
                      min(j + a - r + 1, k + a - r)))
        if i == 0:
            break

    fromY = []
    for i, j, k in dp:
        fromY.append((i + 1, k))
        fromY.append((i, j))

    ans: int = 2e18
    if len(fromX) <= 1:
        ans = min(ans, *map(lambda p: abs(p[0] - fromX[0][0]) + p[1] + fromX[0][1], fromY))
    else:
        k = 1
        for i, j in fromY:
            while i < fromX[k][0] and k + 1 < len(fromX):
                k += 1
            ans = min(ans,
                      abs(i - fromX[k - 1][0]) + j + fromX[k - 1][1],
                      abs(i - fromX[k][0]) + j + fromX[k][1])
    return ans


def main():
    A = sys.stdin.buffer.readlines()
    print(*map(lambda x: solve(*map(int, x.rsplit())), A[1:]), sep='\n')


if __name__ == '__main__':
    main()
0