結果

問題 No.755 Zero-Sum Rectangle
ユーザー H3PO4H3PO4
提出日時 2021-02-10 10:03:37
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
MLE  
実行時間 -
コード長 789 bytes
コンパイル時間 96 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 830,444 KB
最終ジャッジ日時 2024-07-07 19:59:26
合計ジャッジ時間 6,911 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 531 ms
43,884 KB
testcase_01 MLE -
testcase_02 -- -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
evil_1 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
import numpy as np

input = sys.stdin.buffer.readline

N, M = map(int, input().split())
A = np.zeros((M + 1, M + 1), dtype=np.int64)
A[1:, 1:] = np.array([tuple(map(int, input().split())) for _ in range(M)])

acc = A.cumsum(axis=0).cumsum(axis=1)
X = acc[:, None, :, None] + acc[None, :, None, :] - acc[:, None, None, :] - acc[None, :, :, None]
idx = np.argwhere(X == 0).T
idx = idx[:, (idx[0] < idx[1]) & (idx[2] < idx[3])]

imos = np.zeros((M + 1, M + 1), dtype=np.int64)
np.add.at(imos, (idx[0], idx[2]), 1)
np.add.at(imos, (idx[1], idx[3]), 1)
np.subtract.at(imos, (idx[0], idx[3]), 1)
np.subtract.at(imos, (idx[1], idx[2]), 1)

imos = np.array(imos).cumsum(axis=0).cumsum(axis=1)

for _ in range(N):
    x, y = ((int(a) - 1) for a in input().split())
    print(imos[x, y])
0