結果

問題 No.755 Zero-Sum Rectangle
ユーザー titia
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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)])
0