結果

問題 No.325 マンハッタン距離2
ユーザー kimiyuki
提出日時 2016-06-28 20:38:07
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,093 bytes
コンパイル時間 129 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 10,880 KB
最終ジャッジ日時 2024-10-11 22:39:53
合計ジャッジ時間 1,876 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 10 WA * 14
権限があれば一括ダウンロードができます

ソースコード

diff #

#!/usr/bin/env python3
def f(x, y, d): # [0, x) * [0, y)
    assert 0 <= x
    assert 0 <= y
    d = min(d, x + y - 2)
    ans = (d+1)*(d+2)//2
    if x < d:
        dx = d-x+1
        ans -= dx*(dx+1)//2
    if y < d:
        dy = d-y+1
        ans -= dy*(dy+1)//2
    return ans
x1, y1, x2, y2, d = map(int,input().split())
assert x1 < x2
assert y1 < y2
if x2 <= 0:
    x1, x2 = - x2, - x1
if y2 <= 0:
    y1, y2 = - y2, - y1
if x1 <= 0:
    x1, y1 = y1, x1
    x2, y2 = y2, x2
ans = 0
if x1 <= 0:
    assert x1 <= 0 <= x2
    assert y1 <= 0 <= y2
    ans += f(abs(x1),   abs(y1),   d)
    ans += f(abs(x1),   abs(y2)+1, d)
    ans += f(abs(x2)+1, abs(y1),   d)
    ans += f(abs(x2)+1, abs(y2)+1, d)
elif y1 <= 0:
    assert 0 <= x1 < x2
    assert y1 <= 0 <= y2
    ans -= f(x1,   abs(y1),   d)
    ans += f(x2+1, abs(y1),   d)
    ans -= f(x1,   abs(y2)+1, d)
    ans += f(x2+1, abs(y2)+1, d)
    ans -= x2-x1+1
else:
    assert 0 <= x1 < x2
    assert 0 <= y1 < y2
    ans += f(x1,   y1,   d)
    ans -= f(x1,   y2+1, d)
    ans -= f(x2+1, y1,   d)
    ans += f(x2+1, y2+1, d)
print(ans)
0