結果

問題 No.520 プロジェクトオイラーへの招待
ユーザー Min_25Min_25
提出日時 2017-05-28 23:38:22
言語 PyPy2
(7.3.15)
結果
AC  
実行時間 979 ms / 4,000 ms
コード長 848 bytes
コンパイル時間 2,144 ms
コンパイル使用メモリ 77,184 KB
実行使用メモリ 121,700 KB
最終ジャッジ日時 2023-10-21 14:46:19
合計ジャッジ時間 3,977 ms
ジャッジサーバーID
(参考情報)
judge10 / judge12
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 58 ms
76,080 KB
testcase_01 AC 58 ms
76,080 KB
testcase_02 AC 59 ms
76,080 KB
testcase_03 AC 846 ms
104,312 KB
testcase_04 AC 979 ms
121,700 KB
testcase_05 AC 847 ms
104,456 KB
testcase_06 AC 64 ms
76,364 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():
  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)
    ret += rec2(a - 1, b, c - 1)
    return ret

  @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)
    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

  @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)

  T = int(raw_input())
  for _ in range(T):
    a, b, c = map(int, raw_input().split())
    print(calc(a, b, c) % (10 ** 9 + 7))

solve()
0