Advertisement
kosievdmerwe

Untitled

Sep 14th, 2022
1,034
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.63 KB | None | 0 0
  1. MOD = 10**9 + 7
  2.  
  3. calc_primes = lambda n: (
  4.     (
  5.         lambda ans: [
  6.             ans.append(i)
  7.             for i in range(2, n + 1)
  8.             if not [j for j in ans if i % j == 0]
  9.         ] is None or ans
  10.     )([])
  11. )
  12.  
  13. factorial = (lambda n: (
  14.         lambda ans: [ans := (ans * i) % MOD for i in range(1, n + 1)] is None or ans
  15.     )(1)
  16. )
  17.  
  18. class Solution:
  19.     def numPrimeArrangements(self, n: int) -> int:
  20.         return (
  21.             lambda n, num_primes, factorial: factorial(num_primes) * factorial(n - num_primes) % MOD
  22.         )(
  23.             n,
  24.             len(calc_primes(n)),
  25.             factorial
  26.         )
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement