SRM 643 div.2 500 / div.1 250 TheKingsFactorization

TheKingsFactorization.py

# -*- coding: utf-8 -*-
import math,string,itertools,fractions,heapq,collections,re,array,bisect


class TheKingsFactorization:
    def getVector(self, N, primes):
        ans = list(primes)
        n = N
        for p in primes:
            n /= p
        i = 2
        while i ** 3 <= N:
            while n % i == 0:
                n /= i
                ans.append(i)
            i += 1
        if n > 1:
            ans.append(n)
        return tuple(sorted(ans))