ABCのA,B問題埋めをpythonでやってて
AtCoderはnumpyが使えることを知ったのでメモ
ABC 047 B - すぬけ君の塗り絵 2 イージー / Snuke's Coloring 2 (ABC Edit) (200)
(問題文はこちら)
import numpy as np
w, h, n = map(int, raw_input().split())
mp = np.zeros((w, h))
for i in range(n):
xi, yi, ai = map(int, raw_input().split())
if ai == 1: # x < xi
mp[:xi, :] = 1
elif ai == 2: # x > xi
mp[xi:, :] = 1
elif ai == 3: # y < yi
mp[:, :yi] = 1
elif ai == 4: # y > yi
mp[:, yi:] = 1
print(w*h - np.sum(mp))これは便利
→AC
https://beta.atcoder.jp/contests/abc047/submissions/2707715