반응형
https://www.acmicpc.net/problem/6603
6603번: 로또
입력은 여러 개의 테스트 케이스로 이루어져 있다. 각 테스트 케이스는 한 줄로 이루어져 있다. 첫 번째 수는 k (6 < k < 13)이고, 다음 k개 수는 집합 S에 포함되는 수이다. S의 원소는 오름차순으로
www.acmicpc.net
💡IDEA
조합을 사용하는 문제이다.
받아온 k개 길이의 배열 중 6개만큼 선택하는 kC6 경우를 출력하면 된다.
itertools의 combinations 함수를 사용하면 쉽게 6개를 조합한 배열을 돌려주기 때문에 그대로 출력한다.
📌CODE
import sys
input = sys.stdin.readline
from itertools import combinations
while True:
S = list(map(int, input().split()))
k = S[0]
S = S[1:]
if k == 0:
break
for comb in combinations(S, 6):
print(*comb)
print()
반응형
'Algorithm > BOJ' 카테고리의 다른 글
[BOJ][python] 1080. 행렬 (0) | 2022.08.20 |
---|---|
[BOJ][python] 18870. 좌표 압축 (0) | 2022.08.20 |
[BOJ][python] 16234. 인구 이동 (0) | 2022.08.14 |
[BOJ][python] 14891. 톱니바퀴 (0) | 2022.08.13 |
[BOJ][python] 1743. 음식물 피하기 (0) | 2022.08.13 |