結果

問題 No.2509 Beam Shateki
コンテスト
ユーザー miya145592
提出日時 2023-10-21 01:02:56
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,202 ms / 2,000 ms
コード長 2,281 bytes
コンパイル時間 183 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 77,808 KB
最終ジャッジ日時 2024-09-21 01:52:07
合計ジャッジ時間 36,606 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 61
権限があれば一括ダウンロードができます

ソースコード

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 dx, dy in [(0, 1), (1, 0), (1, 1), (-1, 1)]:
    if dx==0:
        rng = range(1, H+1)
    elif dy==0:
        rng = range(1, W+1)
    else:
        rng = range(H+W+4)
    for x in rng:
        sco = 0
        S = set()
        if dx==0:
            nx = x
            ny = 0
        elif dy==0:
            nx = 0
            ny = x
        elif dx==1:
            nx = x-(W+2)
            ny = 0
        else:
            nx = x
            ny = 0
        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 dx2, dy2 in [(0, 1), (1, 0), (1, 1), (-1, 1)]:
            if dx2==0:
                rng2 = range(1, H+1)
            elif dy2==0:
                rng2 = range(1, W+1)
            else:
                rng2 = range(H+W+4)
            for x2 in rng2:
                sco2 = 0
                if dx2==0:
                    nx2 = x2
                    ny2 = 0
                elif dy2==0:
                    nx2 = 0
                    ny2 = x2
                elif dx2==1:
                    nx2 = x2-(W+2)
                    ny2 = 0
                else:
                    nx2 = x2
                    ny2 = 0
                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