高校数学の質問スレ Part438

#!perl
import itertools

# n個のものを3つに分けたときの個数i,j,kのリスト
def divide(n):
return [[i, j-i, n-j] for i in range(n+1) for j in range(i, n+1)]

赤順列, 黒順列, 白順列, 青順列 = divide(4), divide(3), divide(2), divide(1)
箱3 = set()
for 赤,黒,白,青 in itertools.product(赤順列,黒順列,白順列,青順列):
箱 = tuple((赤[i], 黒[i], 白[i], 青[i]) for i in range(3) if 赤[i]+黒[i]+白[i]+青[i] > 0)
if len(箱) == 3: # 空箱なし
箱3.add(tuple(sorted(箱)))

for i in 箱3:
print(i)
print(len(箱3))