結果

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

ソースコード

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+4):
    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+2)
        nx+=dx
        ny+=dy
        if dx==0 or dy==0:
            while 1<=nx<=H and 1<=ny<=W:
                sco += A[nx-1][ny-1]
                S.add((nx, ny))
                nx+=dx
                ny+=dy
        else:
            while 1<=ny<=W:
                if 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, H+W+4):
            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+2)
                nx2+=dx2
                ny2+=dy2
                if dx2==0 or dy2==0:
                    while 1<=nx2<=H and 1<=ny2<=W:
                        if (nx2, ny2) not in S:
                            sco2 += A[nx2-1][ny2-1]
                        nx2+=dx2
                        ny2+=dy2
                else:
                    while 1<=ny2<=W:
                        if 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