結果

問題 No.755 Zero-Sum Rectangle
コンテスト
ユーザー H3PO4
提出日時 2021-02-10 10:03:37
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
MLE  
実行時間 -
コード長 789 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 96 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 830,444 KB
最終ジャッジ日時 2024-07-07 19:59:26
合計ジャッジ時間 6,911 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other MLE * 1 -- * 11
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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