結果
問題 | No.755 Zero-Sum Rectangle |
ユーザー | H3PO4 |
提出日時 | 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 | -- | - |
ソースコード
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])