[백준][6603] 로또

• 문제
  • 6603번: 로또
• 해결

import sys

data = [list(map(int, x.strip().split())) for x in sys.stdin.readlines()]

def go(current_list, remainder_list):
    # 로또 번호를 여섯개 이상 넣은 경우
    if len(current_list) > 6:
        return None

    # 여섯개 추첨
    if len(current_list) == 6:
        print(' '.join([str(x) for x in current_list]))

    # 아직 숫자가 모자르면
    if 0 <= len(current_list) < 6:
        # 남은 숫자가 있다면
        if len(remainder_list):
            for idx, val in enumerate(remainder_list):
                go(current_list + [val], remainder_list[(idx + 1):])

for ind, list_ in enumerate(data):
    if list_[0] == 0:
        exit()
    else:
        if ind != 0:
            print(' ')
        go([], list_[1:])

• 포인트
  • 첫 번호을 기준으로 dynamic programming