[Python]探索(NumPy) ABC165C - Qiita

ABC165C １次方程式の条件なので、NumPy配列を使うことで、計算回数を大きく削減できる。NumPy配列がイテラブルのような役割を果たす。 PyPyでも実行可能で、計算速度を上げることができる。（なぜかメモリ使用量が増える） Python3 1063 ms 106372 KB PyPy3 571 ms 155196 KB import sys import itertools import numpy as np read = sys.stdin.buffer.read readline = sys.stdin.buffer.readline N, M, Q = map(int, readline().split()) A = np.array(list(itertools.combinations_with_replacement(range(1, M + 1), N))) # [