AOJ 4つの整数の和を求めるプログラム

■問題
Sum of 4 Integers

組み合わせの数を求めるプログラムのようです。

AOJ 数列のコンビネーションプログラム」を解くための練習になりそうなので、先にこちらを解くことにします。

import sys
import itertools

def main():
    l = list(range(10))
    
    while True:
        data = sys.stdin.readline().strip()
        if data is None or data == '':
            break

        num = int(data)
        cnt = 0

        for elem in itertools.product(l, repeat=4):
            sum = 0
            for i in elem:
                sum += i

            if sum == num:
                cnt += 1

        print(cnt)

if __name__ == '__main__':
    main()
Pythonプログラミング物語 © 2016 Frontier Theme