結果

問題 No.2509 Beam Shateki
ユーザー miya145592
提出日時 2023-10-21 00:26:36
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,319 bytes
コンパイル時間 322 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 77,424 KB
最終ジャッジ日時 2024-09-21 00:59:36
合計ジャッジ時間 20,579 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 56 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
H, W = map(int, input().split())
A = [list(map(int, input().split())) for _ in range(H)]
ans = 0
for x in range(H+W+3):
    if 0<=x<=H+1:
        dir = [(0, 1), (1, 1), (-1, 1)]
    else:
        dir = [(1, 0), (1, 1), (1, -1)]
    for dx, dy in dir:
        sco = 0
        S = set()
        if 0<=x<=H+1:
            nx = x
            ny = 0
        else:
            nx = 0
            ny = x-(H+1)
        nx+=dx
        ny+=dy
        while 1<=nx<=H and 1<=ny<=W:
            sco += A[nx-1][ny-1]
            S.add((nx, ny))
            nx+=dx
            ny+=dy

        for x2 in range(x+1, H+W+3):
            if 0<=x2<=H+1:
                dir2 = [(0, 1), (1, 1), (-1, 1)]
            else:
                dir2 = [(1, 0), (1, 1), (1, -1)]
            for dx2, dy2 in dir2:
                sco2 = 0
                if 0<=x2<=H+1:
                    nx2 = x2
                    ny2 = 0
                else:
                    nx2 = 0
                    ny2 = x2-(H+1)
                nx2+=dx2
                ny2+=dy2
                while 1<=nx2<=H and 1<=ny2<=W:
                    if (nx2, ny2) not in S:
                        sco2 += A[nx2-1][ny2-1]
                    nx2+=dx2
                    ny2+=dy2

                ans = max(ans, sco+sco2)
print(ans)
0