結果

問題 No.325 マンハッタン距離2
ユーザー rpy3cpp
提出日時 2015-12-19 11:42:23
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 30 ms / 1,000 ms
コード長 2,083 bytes
コンパイル時間 90 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 11,008 KB
最終ジャッジ日時 2024-09-16 08:54:47
合計ジャッジ時間 1,563 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

x1, y1, x2, y2, d = map(int, input().split())
def solve(x1, y1, x2, y2, d):
    x1, y1, x2, y2 = relocate(x1, y1, x2, y2)
    if x1 < x2 or x1 < 0 or y1 < 0 or y1 < y2:
        raise RuntimeError('relocate failed. x1:{}, y1:{}, x2:{}, y2:{}'.format(x1, y1, x2, y2))
    if x2 > 0:
        if y2 > 0:
            return f(x1, y1) - f(x2 - 1, y1) - f(x1, y2 - 1) + f(x2 - 1, y2 - 1)
        elif y2 == 0:
            return f(x1, y1) - f(x2 - 1, y1)
        else:
            return f(x1, y1) - f(x2 - 1, y1) + f(x1, -y2) - f(x2 - 1, -y2) - min(x1, d) + min(x2 - 1, d)
    elif x2 == 0:
        if y2 > 0:
            return f(x1, y1) - f(x1, y2 - 1)
        elif y2 == 0:
            return f(x1, y1)
        else:
            return f(x1, y1) + f(x1, -y2) - min(x1, d) - 1
    else:
        if y2 > 0:
            xx1 = max(y1, y2)
            yy1 = max(x1, x2)
            xx2 = min(y1, y2)
            yy2 = min(x1, x2)
            return f(xx1, yy1) - f(xx2 - 1, yy1) + f(xx1, -yy2) - f(xx2 - 1, -yy2) - min(xx1, d) + min(xx2 - 1, d)
        elif y2 == 0:
            return f(x1, y1) + f(-x2, y1) - min(y1, d) - 1
        else:
            total = f(x1, y1) + f(-x2, -y2) + f(x1, -y2) + f(-x2, y1)
            vertical_double_counts = min(y1, d) - max(y2 - 1, - d - 1)
            horizontal_double_counts = min(x1, d) - max(x2 - 1, - d - 1)
            return total - vertical_double_counts - horizontal_double_counts - 1
            
def relocate(x1, y1, x2, y2):
    if x2 > 0:
        if y2 > 0:
            return x2, y2, x1, y1
        else:
            return x2, -y1, x1, -y2        
    else:
        if y2 > 0:
            return -x1, y2, -x2, y1
        else:
            return -x1, -y1, -x2, -y2
def f(x, y):
    if x < 0 or y < 0:
        raise RuntimeError('x and y must be non-negative. x:{}, y:{}'.format(x, y))
    if x + y <= d:
        return (x + 1) * (y + 1)
    ydif = max(0, d - y)
    xdif = max(0, d - x)
    return area(d + 1) - area(ydif) - area(xdif)
        
def area(size):
    return size * (size + 1) // 2    
print(solve(x1, y1, x2, y2, d))
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0