Submission #214360

---

Source Code Expand

#!/usr/bin/env python2.7

import sys
from collections import namedtuple

from cStringIO import StringIO
import unittest
import cProfile

def main():
    N, H = (int(x) for x in sys.stdin.readline().split())
    A, B, C, D, E = (int(x) for x in sys.stdin.readline().split())
    print solve(N, H, A, B, C, D, E)

def solve(N, H, A, B, C, D, E):
    assert N <= 10

    if N == 1:
        if H > E:
            return 0
        else:
            return C
    else:
        a = solve(N-1, H+B, A, B, C, D, E) + A
        b = solve(N-1, H+D, A, B, C, D, E) + C

        if H > E:
            c = solve(N-1, H-E, A, B, C, D, E)
            return min([a, b, c])
        else:
            return min([a, b])

    
class Test(unittest.TestCase):

    @staticmethod
    def tryone(indata):
        sys.stdin = StringIO(indata)
        out = sys.stdout = StringIO()
        main()
        return out.getvalue()

    def test51(self):
        self.assertEqual(solve(4, 5, 100, 4, 60, 1, 4), 160)

    def test52(self):
        self.assertEqual(solve(10, 1, 5000, 2, 2000, 1, 300), 20000)

    def test53(self):
        self.assertEqual(solve(9, 23, 170, 8, 120, 5, 12), 650)

    def test54(self):
        self.assertEqual(solve(653, 314159, 6728, 123456, 5141, 41928, 222222), 2818162)

    def test90(self):
        self.assertEqual(self.tryone("""\
4 5
100 4 60 1 4
"""), """\
160
""")

if __name__ == '__main__':
    if len(sys.argv) > 1:
        print "_/" * 30 + str(sys.argv)
        if sys.argv[1] == '-p':
            sys.argv.pop(1)
            cProfile.run("unittest.main(exit=False, failfast=True)", sort='time')
        else:
            unittest.main()
    else:
        main()

Submission Info

Judge Result