Advertisement
faubiguy

bracket.py

Nov 5th, 2014
219
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.44 KB | None | 0 0
  1. #!/usr/bin/python3
  2. import sys
  3.  
  4. def primes(n):
  5.     candidates = range(2,n+1)
  6.  
  7.     primes = []
  8.  
  9.     while candidates:
  10.         prime = candidates[0]
  11.         primes.append(prime)
  12.         candidates = list(filter(lambda x:x%prime, candidates[1:]))
  13.    
  14.     return len(primes), n == 1 or primes[-1] == n
  15.  
  16. def factor(n):
  17.     factors = []
  18.     i = 2
  19.     while i * i <= n:
  20.         while n % i == 0:
  21.             n = n // i
  22.             factors.append(i)
  23.         i = i + 1
  24.     factors.append(n)
  25.     return [factor for factor in factors if factor != 1]
  26.    
  27. def get_brackets(n, include_primality=False):
  28.     symbols = [n]
  29.    
  30.     n_is_prime = None
  31.  
  32.     while True:
  33.         for index, symbol in enumerate(symbols):
  34.             if type(symbol) == int:
  35.                 if symbol == 1:
  36.                     symbols[index] = ''
  37.                     break
  38.                 prime_count, is_prime = primes(symbol)
  39.                 if n_is_prime is None:
  40.                     n_is_prime = is_prime
  41.                 #if is_prime:
  42.                     #p_arg = Number(symbol.prime_count())
  43.                     #symbols = symbols[:index] + ['<', p_arg, '>'] + symbols[index+1:]
  44.                     #break
  45.                 #else:
  46.                     #symbols = symbols[:index] + [Number(n) for n in factor(symbol.value)] + symbols[index+1:]
  47.                     #break
  48.                 symbols = symbols[:index] + (
  49.                     ['<', prime_count, '>']
  50.                     if is_prime
  51.                     else factor(symbol)
  52.                 ) + symbols[index+1:]
  53.                 break
  54.         else:
  55.             break
  56.     brackets = ''.join(symbols)
  57.     return (brackets, n_is_prime) if include_primality else brackets
  58.  
  59. if __name__ == '__main__'
  60.     n = int(sys.argv[1] if len(sys.argv) > 1 else sys.stdin.read())
  61.     print(get_brackets(n))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement