44-xxubin04

[xxubin04]

🔗 문제 링크

6603번: 로또💰

✔️ 소요된 시간

30분

✨ 수도 코드

1. 문제 이해

입력받는 집합의 원소들 중 6개로 만들 수 있는 모든 조합을 출력하는 문제이다.
출력할 때, 각 테스트 케이스 사이에 빈 줄이 하나 있어야 한다..!

2. combinations

from itertools import combinations
input = open(0).readline

while (numComb := list(map(int, input().split()))) != [0]:
    for i in combinations(numComb[1:], 6):
        print(" " .join(str(j) for j in list(i)))
    print()

입력받은 리스트가 [0] 즉, 0이 입력된 경우에는 프로그램을 끝낸다.
[0]이 아니라면, 만들 수 있는 조합을 하나씩 출력해주면 된다...!
combinations를 사용하면 매우 간단하게 풀 수 있다.

3. 백트래킹

input = open(0).readline 

def backTracing(numComb, comb):
    if len(comb) == 6:
        print(" ".join(map(str, comb)))
        return
    for i in numComb[1:]:
        if i not in comb and i > comb[-1]:
            comb.append(i)
            backTracing(numComb, comb)
            comb.pop()
            
while (numComb := list(map(int, input().split()))) != [0]:
    comb = []
    for start in range(1, numComb[0]-4):
        backTracing(numComb, [numComb[start]])
    print()

0이 입력되지 않은 경우에는 6개 숫자의 조합에서 제일 앞에 올 수 있는 수가 range(1, numComb[0]-4으로 집합 S의 개수에서 5를 뺀 개수만큼 올 수 있다.

만약, 집합 S = {1,5,7,13,15,20, 33, 40, 45}라면 제일 마지막 조합으로 나오는 것이 {13, 15, 20, 33, 40, 45}이다. 그러므로 조합의 제일 앞에 올 수 있는 수는 1, 5, 7, 13으로 4개이므로 집합 S의 개수 9에서 5를 뺀 개수가 같은 것이다.

[numComb[start]] 으로 조합의 맨 앞 수를 지정해주지 않고 다음의 코드 부분을 실행하면 index 오류가 나서 조합의 맨 앞에 올 수 있는 경우를 먼저 정해주고 백트래킹하는 방법을 생각했다.

    for i in numComb[1:]:
        if i not in comb and i > comb[-1]:
            comb.append(i)
            backTracing(numComb, comb)
            comb.pop()

📚 새롭게 알게된 내용

백트래킹이 아직 익숙하지 않아서 여러 문제 풀어봐야 할 것 같네요..🤔
백트래킹으로 푼 방법이 출력초과 문제가 해결되지 않았다가 PR 올리면서 해결해서 코드파일은 못 올렸습니당

[9kyo-hwang]

이 문제 익숙하다 싶었는데 옆동네 돌다가 풀었던 문제네요...
그때도 파이썬 combinations 개사기! 했던 기억이 ㅋㅋ
직접 백트래킹 구현해서 풀어본 거 완전 굿 👍

from itertools import combinations
input = open(0).readline

while (S := list(map(str, input().split()))) and S[0] != 0:
    for comb in combinations(S[1:], 6):
        print(' '.join(comb))
    print()

[gjsk132]

늦어서 백트래킹 연속 2문제..! 그래서 그런지 나름 스피드하게 코드가 짜졌습니다🥲

약간 다른 점은 backtracking에 넣는 매개변수를 comb 리스트에 들어있는 숫자의 수와 마지막 수(가장 큰 수)로 지정했습니다!

코드!

input = open(0).readline

def backtracking(cnt, last):
    if cnt == 6:
        print(*comb)
        return
    
    for num in lotto:
        if num < last or num in comb:
            continue
        comb.append(num)
        backtracking(cnt+1, num)
        comb.pop()

while True:
    lotto = list(map(int,input().split()))

    if lotto.pop(0) == 0:
        break

    lotto.sort()
    comb = []

    backtracking(0, 0)
    print()