Submission #169342

---

Source Code Expand

#!/usr/bin/env python2.7

import sys
from collections import defaultdict

from cStringIO import StringIO
import unittest
import cProfile

def main():
    N, = (int(x) for x in sys.stdin.readline().split())
    S = []
    for _ in range(N):
        s, = (str(x) for x in sys.stdin.readline().split())
        S.append(s)
    print solve(S)

def solve(S):
    d = defaultdict(int)
    for s in S:
        d[s] += 1
    m = max(d.values())
    for s in S:
        if d[s] == m:
            return s
    
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(["taro", "jiro", "taro", "saburo"]), "taro")
        self.assertEqual(solve(["takahashikun"]), "takahashikun")
        self.assertEqual(solve(["a", "b", "c", "c", "b", "c", "b", "d", "e"]), "b") # my solve() returns first match

    def test90(self):
        self.assertEqual(self.tryone("""\
4
taro
jiro
taro
saburo
"""), """taro\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