[백준 3049][Python] 다각형의 대각선

문제 링크 https://www.acmicpc.net/problem/3049

다각형에서 대각선의 교접의 갯수를 구하는 식은 nC4 이다.
그리고 반례인 삼각형의 경우 따로 처리해 주어야한다.

풀이

import operator as op
from functools import reduce


def nc4(n: int):
    r = min(4, n-4)
    numer = reduce(op.mul, range(n, n-r, -1), 1)
    denom = reduce(op.mul, range(1, r+1), 1)
    return numer // denom


n = int(input())
if n == 3:
    print(0)
else:
    print(nc4(n))