Submission #154813

---

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):
    for c in str(N):
        if c == '3':
            return 'YES'
    if N % 3 == 0:
        return 'YES'
    else:
        return 'NO'
    
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), "NO")
        self.assertEqual(solve(2), "NO")
        self.assertEqual(solve(3), "YES")
        self.assertEqual(solve(4), "NO")
        self.assertEqual(solve(31), "YES")
        self.assertEqual(solve(32), "YES")
        self.assertEqual(solve(33), "YES")

    def test91(self):
        self.assertEqual(self.tryone("""\
2
"""), """NO\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