[백준 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))