結果

問題 No.2509 Beam Shateki
ユーザー ntuda
提出日時 2023-10-28 19:13:55
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,140 bytes
コンパイル時間 233 ms
コンパイル使用メモリ 82,288 KB
実行使用メモリ 84,776 KB
最終ジャッジ日時 2024-09-25 16:33:12
合計ジャッジ時間 11,229 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10 TLE * 1 -- * 50
権限があれば一括ダウンロードができます

ソースコード

diff # 

H, W = map(int, input().split())
A = [[0] * (W + 2)]
A += [[0] + list(map(int, input().split())) + [0] for _ in range(H)]
A += [[0] * (W + 2)]
H += 2
W += 2

N = (H + W) * 9 * 2
D = []
for i in range(-1, 2):
    for j in range(-1, 2):
        D.append((i, j))
ans = 0

def HWC(x):
    if x < H * 9:
        return (x // 9, 0)
    x -= H * 9
    if x < H * 9:
        return (x // 9, W - 1)
    x -= H * 9
    if x < W * 9:
        return (0, x // 9)
    x -= W * 9
    return (H - 1, x // 9)

for i in range(N-1):
    x, y = HWC(i)
    cnt1 = 0
    P = set()
    dx, dy = D[i % 9]
    x += dx
    y += dy
    while 0 <= x < H and 0 <= y < W and A[x][y] > 0:
        cnt1 += A[x][y]
        P.add((x, y))
        x += dx
        y += dy
    for j in range(i + 1, N):
        x, y = HWC(j)
        cnt2 = 0
        dx, dy = D[j % 9]
        x += dx
        y += dy
        while 0 <= x < H and 0 <= y < W and A[x][y] > 0:
            if (x, y) not in P:
                cnt2 += A[x][y]
            x += dx
            y += dy
        ans = max(ans, cnt1 + cnt2)
        #print(i, HWC(i), D[i % 9], j, HWC(j), D[j % 9], cnt1, cnt2)
print(ans)
0