結果
問題 |
No.755 Zero-Sum Rectangle
|
ユーザー |
![]() |
提出日時 | 2025-02-10 03:25:42 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,601 ms / 2,000 ms |
コード長 | 1,509 bytes |
コンパイル時間 | 410 ms |
コンパイル使用メモリ | 82,240 KB |
実行使用メモリ | 77,284 KB |
最終ジャッジ日時 | 2025-02-10 03:25:57 |
合計ジャッジ時間 | 13,912 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 12 |
ソースコード
import sys input = sys.stdin.readline def idx(i, j, m): return i * (m + 2) + j n, m = map(int, input().split()) A = [list(map(int, input().split())) for _ in range(m)] # Flattened 1D array for S S = [0] * ((m + 2) * (m + 2)) # Compute prefix sums for i in range(m): for j in range(m): S[idx(i, j, m)] = A[i][j] for i in range(m): for j in range(1, m): S[idx(i, j, m)] += S[idx(i, j - 1, m)] for i in range(1, m): for j in range(m): S[idx(i, j, m)] += S[idx(i - 1, j, m)] def rectsum(x, y, z, w, m): return ( S[idx(z, w, m)] - (S[idx(z, y - 1, m)] if y > 0 else 0) - (S[idx(x - 1, w, m)] if x > 0 else 0) + (S[idx(x - 1, y - 1, m)] if x > 0 and y > 0 else 0) ) # Flattened 1D array for LS LS = [0] * ((m + 2) * (m + 2)) # Compute LS updates for i in range(m): for j in range(i, m): for k in range(m): for l in range(k, m): if rectsum(i, k, j, l, m) == 0: LS[idx(i, k, m)] += 1 LS[idx(i, l + 1, m)] -= 1 LS[idx(j + 1, k, m)] -= 1 LS[idx(j + 1, l + 1, m)] += 1 # Apply prefix sums on LS for i in range(1, m + 2): for j in range(m + 2): LS[idx(i, j, m)] += LS[idx(i - 1, j, m)] for i in range(m + 2): for j in range(1, m + 2): LS[idx(i, j, m)] += LS[idx(i, j - 1, m)] # Answer queries for _ in range(n): x, y = map(int, input().split()) print(LS[idx(x - 1, y - 1, m)])