結果
問題 | No.2509 Beam Shateki |
ユーザー |
|
提出日時 | 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 |
ソースコード
import sysinput = sys.stdin.readlineH, W = map(int, input().split())A = [list(map(int, input().split())) for _ in range(H)]ans = 0for 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 = 0S = set()if dx==0:nx = xny = 0elif dy==0:nx = 0ny = xelif dx==1:nx = x-(W+2)ny = 0else:nx = xny = 0nx+=dxny+=dyif dx==0 or dy==0:while 1<=nx<=H and 1<=ny<=W:sco += A[nx-1][ny-1]S.add((nx, ny))nx+=dxny+=dyelse:while 1<=ny<=W:if 1<=nx<=H and 1<=ny<=W:sco += A[nx-1][ny-1]S.add((nx, ny))nx+=dxny+=dyfor 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 = 0if dx2==0:nx2 = x2ny2 = 0elif dy2==0:nx2 = 0ny2 = x2elif dx2==1:nx2 = x2-(W+2)ny2 = 0else:nx2 = x2ny2 = 0nx2+=dx2ny2+=dy2if 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+=dx2ny2+=dy2else: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+=dx2ny2+=dy2ans = max(ans, sco+sco2)print(ans)