Programa em Python 22

Dados dois números naturais n e p, sendo n > p, escreva um programa para calcular a
combinação de n elementos tomados p a p.

Resolução:

def verifica_p(n, p):
    if (p > n):
        print("Entrada invalida!")


def calcula_combinacao(n, p):
    n_fat = 1
    for a in range(1, n + 1):
        n_fat = a * n_fat

    for a in range(1, p + 1):
        p_fat = 1
        p_fat = a * p_fat

    np = n - p
    for a in range(1, np + 1):

        np_fat = 1
        np_fat = a * np_fat


    comb = n_fat / (p_fat * np_fat)
    print("A combinacao de ", n, " elementos, tomados  ", p, " a ", p, " e igual a ", comb)


n = int(input("Digite um numero n: "))
p = int(input("Digite um numero p (menor que n): "))

verifica_p(n, p)

calcula_combinacao(n, p)