sopj 7190. Guess the Number | GUESSTHE

guessthe.py

from sys import stdin

def main():
    def gcd(x, y):
        while y:
            x, y = y, x % y
        return x

    factors = ((), (), (), (2,), (), (3, 2), (), (4, 2), (3,), (5, 2), (), (6, 4, 3, 2), (), (7, 2), (5, 3), (8, 4, 2), (), (9, 6, 3, 2), (), (10, 5, 4, 2))

    def the_number(s):
        if s[0] == 'N':
            return '-1'
        number = 1
        for i, x in enumerate(s):
            if x == 'Y':
                for y in factors[i]:
                    if s[y - 1] == 'N':
                        return '-1'
                number *= (i + 1) // gcd(number, i + 1)

        for i, x in enumerate(s):
            if x == 'N' and number % (i + 1) == 0:
                return '-1'
        return str(number)

    lines = stdin.read().split()
    lines.pop()
    out = []
    for s in lines:
        out.append(the_number(s))
    print '\n'.join(out)



main()