[백준] 15439 : 베라의 패션 (Python/파이썬)

문제 링크

https://www.acmicpc.net/problem/15439

문제

베라는 상의 N 벌과 하의 N 벌이 있다. i 번째 상의와 i 번째 하의는 모두 색상 i를 가진다. 
N 개의 색상은 모두 서로 다르다.

상의와 하의가 서로 다른 색상인 조합은 총 몇 가지일까?

입력

입력은 아래와 같이 주어진다.

N

출력

상의와 하의가 서로 다른 색상인 조합의 가짓수를 출력한다.

예제

입력

5

출력

20

---

정답 및 풀이

import sys

input = sys.stdin.readline

n = int(input())
cnt = 0
for i in range(1, n + 1):
    for j in range(1, n + 1):
        if i != j:
            cnt += 1
print(cnt)

1. import sys
   • 빠른 입력을 사용하기 위해 sys 모듈을 불러온다.
2. input = sys.stdin.readline
   • input() 대신 더 빠른 입력 방식으로 설정한다.
3. n = int(input())
   • 옷의 색깔 개수 n을 입력받는다.
4. cnt = 0
   • 서로 다른 조합의 개수를 세기 위한 변수 초기화.
5. for i in range(1, n + 1):
   • 첫 번째 옷(상의) 색깔을 1부터 n까지 반복한다.
6. for j in range(1, n + 1):
   • 두 번째 옷(하의) 색깔을 1부터 n까지 반복한다.
7. if i != j:
   • 상의와 하의 색깔이 서로 다를 때만 경우의 수로 인정.
8. cnt += 1
   • 서로 다른 색 조합 한 개를 경우의 수로 추가한다.
9. print(cnt)
   • 최종적으로 계산된 조합 개수를 출력한다.

다른 풀이

import sys
input = sys.stdin.readline

n = int(input())
print(n * (n - 1))

---