結果

問題 No.520 プロジェクトオイラーへの招待
ユーザー Min_25Min_25
提出日時 2017-05-28 23:42:45
言語 PyPy2
(7.3.15)
結果
AC  
実行時間 857 ms / 4,000 ms
コード長 885 bytes
コンパイル時間 1,774 ms
コンパイル使用メモリ 76,416 KB
実行使用メモリ 96,412 KB
最終ジャッジ日時 2024-09-21 16:00:40
合計ジャッジ時間 3,422 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 78 ms
75,264 KB
testcase_01 AC 80 ms
75,308 KB
testcase_02 AC 76 ms
75,136 KB
testcase_03 AC 732 ms
90,152 KB
testcase_04 AC 857 ms
96,412 KB
testcase_05 AC 758 ms
88,704 KB
testcase_06 AC 80 ms
75,904 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

def memoize(f):
  cache = {}
  def helper(*args):
    if args not in cache:
      cache[args] = f(*args)
    return cache[args]
  return helper

def solve():
  mod = 10 ** 9 + 7

  def calc(a, b, c):
    ret = 0
    for i in range(1, b + 1):
      ret += rec_tri(a - 1, i - 1) * rec_tri(c, b - i) % mod
    ret += rec2(a - 1, b, c - 1)
    return ret % mod

  @memoize
  def rec2(a, b, c):
    ret = 0
    for i in range(1, b + 1):
      ret += rec_tri(a, i - 1) * rec_tri(c, b - i) % mod
    ret += rec_tri(b, c) if a == 0 else rec2(a - 1, b, c)
    ret += rec_tri(a, b) if c == 0 else rec2(a, b, c - 1)
    return ret % mod

  @memoize
  def rec_tri(a, b):
    ret = 0
    if a == 0 or b == 0: return 1
    return (rec_tri(a - 1, b) + rec_tri(a, b - 1)) % mod

  T = int(raw_input())
  for _ in range(T):
    a, b, c = map(int, raw_input().split())
    print(calc(a, b, c))

solve()
0