[구현/수학] 백준 2921 도미노 - 파이썬(Python)

 

[ Contents ]

 

 

1. 문제 (링크 참조)

2921번: 도미노

 

 

 

2. 문제 풀이

 크기가 N인 도미노 세트에 점이 몇 개 있는지 구하는 문제입니다.

 

		○		○	○○
	○	○	○○	○○	○○

 크기가 2인 도미노 세트는 12개입니다. 총 6가지 도미노가 나오며, (윗 칸, 아랫 칸)으로 표시하면 다음과 같습니다.

 (0, 0), (0, 1), (1, 1), (0, 2), (1, 2), (2, 2)

 즉, 0, 1, 2 중에서 2개를 뽑을 중복조합의 수와 같습니다.

 

 

 

3. 코드

from itertools import combinations_with_replacement

# 입력
n = int(input())

# 중복조합
res = 0
for i in combinations_with_replacement(range(n+1), 2):
    res += sum(i)
print(res)

 점의 개수는 0 ~ n개로 총 n+1가지 중 2개를 뽑는 중복조합입니다. 각 중복조합의 점의 개수를 더한 뒤 출력합니다.