Triângulo de Floyd

O triângulo de Floyd é uma matriz triangular de números naturais com ângulo reto. Ele é definido preenchendo as linhas do triângulo com números consecutivos, começando com 1 no canto superior esquerdo:

Escreva uma função que receba um inteiro n e retorne as linhas do triângulo de Floyd como uma lista de listas. Sua função deve retornar uma de duas possibilidades:

• Se um valor for passado para n_row, retorne as primeiras n linhas.
• Se um valor for passado para up_to, retorne todas as linhas até, e incluindo, a linha que contém n.

Espera-se que um argumento seja passado para um dos parâmetros, mas não para ambos.

Exemplos

floyd(up_to = 5) ➞ [[1], [2, 3], [4, 5, 6]]
# The third row contains a 5.

floyd(n_row = 5) ➞[[1], [2, 3], [4, 5, 6], [7, 8, 9, 10], [11, 12, 13, 14, 15]]
# Returns the first five rows.

floyd(up_to = 10) ➞ [[1], [2, 3], [4, 5, 6], [7, 8, 9, 10]]

floyd(n_row = 1) ➞[[1]]

Notas

Dica: Você pode definir n_row a partir de up_to usando a sequência de números triangulares. Ou seja, n_row deve ser x na equação (x*(x+1))/2 = up_to arredondada para cima. Depois, você só precisa escrever uma função para n_row.