結果

問題 No.755 Zero-Sum Rectangle
ユーザー H3PO4H3PO4
提出日時 2021-02-10 10:07:46
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
MLE  
実行時間 -
コード長 842 bytes
コンパイル時間 71 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 830,528 KB
最終ジャッジ日時 2024-07-07 20:02:30
合計ジャッジ時間 5,934 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 447 ms
44,524 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)
del A
idx = np.argwhere(acc[:, None, :, None]
                  + acc[None, :, None, :]
                  - acc[:, None, None, :]
                  - acc[None, :, :, None] == 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