〈メモ〉AtCoderでnumpyを使う

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