結果

問題 No.755 Zero-Sum Rectangle
コンテスト
ユーザー H3PO4
提出日時 2021-02-10 10:07:46
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
MLE  
実行時間 -
コード長 842 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 71 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 830,528 KB
最終ジャッジ日時 2024-07-07 20:02:30
合計ジャッジ時間 5,934 ms
ジャッジサーバーID
（参考情報） 		judge3 / 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)
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