012: 四角い三角関係

ヒント / 解答

難易度: ☆☆

問題

平方数とは、ある正の整数 m を使って m * m と表せる数です。 たとえば、1、4、9、16 は平方数です。

三角数とは、ある正の整数 k を使って k * (k + 1) // 2 と表せる数です。 たとえば、1、3、6、10 は三角数です。

平方数でも三角数でもある正の整数を、平方三角数と呼びます。 平方三角数を小さい順に10個返す関数 square_triangular_numbers を書いてください。

制約

• 引数 count は1以上10以下の整数です。
• 関数は、小さい順に並んだ平方三角数のリストを返します。

例

>>> square_triangular_numbers(1)
[1]
>>> square_triangular_numbers(3)
[1, 36, 1225]

目標

>>> square_triangular_numbers(10) == [
...     1,
...     36,
...     1225,
...     41616,
...     1413721,
...     48024900,
...     1631432881,
...     55420693056,
...     1882672131025,
...     63955431761796,
... ]
True

参考

• 平方三角数
• OEIS A001110