Фигуры из точек

Многоугольные числа можно представить как количество точек, из которых собирают треугольники, квадраты и другие фигуры. Даны номер элемента num и количество сторон sides. Верни num-е многоугольное число для фигуры с таким числом сторон. Например, для квадрата это будут квадратные числа.

Примеры

Ввод
0, 3
Вывод
0
Ввод
3, 3
Вывод
6
Ввод
5, 4
Вывод
25

Нужно реализовать

def polygonal_num(num: int, sides: int) -> int:

Теги

math
PostgreSQLv16

Здесь будет результат выполнения кода

Видимые тесты
1
Ввод
0, 3
Ожидаемый0
2
Ввод
3, 3
Ожидаемый6
3
Ввод
5, 4
Ожидаемый25
4
Ввод
2, 5
Ожидаемый5
5
Ввод
4, 5
Ожидаемый22
6
Ввод
3, 7
Ожидаемый18
Фокус-радио
На паузе · SomaFM · Fluid