結果

問題 No.117 組み合わせの数
コンテスト
ユーザー nebukuro09
提出日時 2016-09-21 17:13:54
言語 PyPy2
(7.3.15)
結果
TLE  
実行時間 -
コード長 991 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 380 ms
コンパイル使用メモリ 77,056 KB
実行使用メモリ 218,316 KB
最終ジャッジ日時 2024-11-17 10:29:23
合計ジャッジ時間 12,837 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import re

MOD = 10**9+7

def ncr(n, r):
    if n - r < r:
        r = n-r
    if r == 0:
        return 1
    if r == 1:
        return n

    numerator = [0 for i in range(r)]
    denominator = [0 for i in range(r)]

    for k in range(r):
        numerator[k] = n - r + k + 1
        denominator[k] = k + 1
    
    for p in range(2, r+1):
        pivot = denominator[p-1]
        if pivot > 1:
            offset = (n-r) % p
        for k in range(p-1, r, p):
            numerator[k-offset] /= pivot
            denominator[k] /= pivot

    result = 1
    for k in range(r):
        if numerator[k] > 1:
            result = (result*numerator[k]) % MOD

    return result

def npr(n, r):
    ans = 1
    for i in xrange(r):
        ans = (ans*(n-i))%MOD
    return ans

for i in xrange(input()):
    q = raw_input()
    n, r = map(int, re.findall('\d+', q))
    if q[0] == 'C':
        print ncr(n, r)
    elif q[0] == 'P':
        print npr(n, r)
    else:
        print ncr(n+r-1, r)
0