結果

問題 No.520 プロジェクトオイラーへの招待
コンテスト
ユーザー Min_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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 7
権限があれば一括ダウンロードができます

ソースコード

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