Submission #154969

---

Source Code Expand

#!/usr/bin/env python2.7

import sys
from collections import namedtuple, defaultdict

from cStringIO import StringIO
import unittest
import cProfile

def main():
    n, = (int(x) for x in sys.stdin.readline().split())
    print solve(n)

def solve(n):
    if n == 1:
        return 0
    if n == 2:
        return 0
    if n == 3:
        return 1
    a, b, c = 0, 0, 1
    for _ in xrange(n-3):
        a, b, c = b, c, (a + b + c) % 10007
    return c
    
class Test(unittest.TestCase):

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

    def test50(self):
        self.assertEqual(solve(1), 0)
        self.assertEqual(solve(2), 0)
        self.assertEqual(solve(3), 1)
        self.assertEqual(solve(4), 1)
        self.assertEqual(solve(5), 2)
        self.assertEqual(solve(6), 4)
        self.assertEqual(solve(7), 7)
        self.assertEqual(solve(100000), 7927)

    def test91(self):
        self.assertEqual(self.tryone("""\
7
"""), """7\n""")

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