[백준/Python/파이썬] 6603번 로또

1. 문제

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

 

2. 풀이

전체 코드

from itertools import combinations

while True:
    num = input()
    if len(num)==1 and int(num) == 0:
        break
    else:
        num = num.split()

    k = num[0]
    num.pop(0)

    result = list(combinations(num, 6))
    for i in range(len(result)):
        for j in range(len(result[i])):
            print(result[i][j], end=' ')
        print()
    print()

 

1. 입력 받기

k와 로또 숫자 k개가 한 줄에 입력되므로, 나눠서 변수에 담는 동시에 0만 입력됐을 경우에는 프로그램을 종료해야 함.

따라서 num=input()으로 받은 후, 길이가 1이고 숫자로 변환했을 때 0이면 while을 탈출(break)한다는 코드를 while 반복 상단에 작성.

 

while 조건이 True로 break를 하지 않는 이상 계속 반복될 것이기 때문에, 코드를 실행하면 바로 입력받게 됨.

0이 입력된게 아닐 경우(=k와 로또 숫자 k개가 한 줄에 입력된 경우), num=num.split()을 통해띄어쓰기 기준으로 분리하여 리스트를 만듦.

 

2. 변수 정리

만들어진 num 리스트는 첫 번째 요소(index=0)만 k이고 나머지가 로또 숫자임.

따라서 k는 num[0]을 통해 따로 담고, pop()을 통해 num 리스트에 로또 숫자만 남기기.

 

3. 조합 이용

 

문제에서 말한 로또 번호 집합에서 고를 수 있는 경우의 수는 위의 공식과 같음

파이썬에는 조합을 쉽게 구현할 수 있는 라이브러리가 있음

from itertools import combinations를 통해 combinations 메서드를 사용할 수 있음

result = list(combinations(num, 6))를 통해 result에 num리스트에서 6개를 뽑는 가능한 케이스를 모두 담기

 

4. 출력

여기서 그냥 result를 출력해버리면 아래와 같이 나온다.

[('1', '2', '3', '4', '5', '6'), 
('1', '2', '3', '4', '5', '7'), 
('1', '2', '3', '4', '6', '7'), 
('1', '2', '3', '5', '6', '7'), 
('1', '2', '4', '5', '6', '7'), 
('1', '3', '4', '5', '6', '7'), 
('2', '3', '4', '5', '6', '7')]

 

따라서 이를 문제에서 원한 포맷으로 출력하기 위한 코드를 작성한다.