[백준] [Python] 15655번 N과 M (6) - 백트래킹 - [대표예제]

 

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

15655번: N과 M (6)

 

 

Python

n, m = map(int, input().split())
arr = list(map(int, input().split()))
arr.sort()

nums = []

def back(k, idx):
    if k == m:
        print(' '.join(map(str, nums)))
        return
    for i in range(idx, n):
            nums.append(arr[i])
            back(k+1, i+1)
            nums.pop()
            
back(0, 0)

 

 

오답

문자열을 그대로 정렬하면 사전순 배열이 되어 오답

꼭 숫자로 만들어 정렬한 뒤, join할 때 str로 바꿔주자

import sys
n, m = map(int, sys.stdin.readline().split())

####
arr =  sys.stdin.readline().split()
arr.sort()
####

visit = [0]*n
nums = []
def back(k,idx):
    if k==m:
        print(" ".join(nums))
        return
    for i in range(idx,n):
            nums.append(arr[i])
            back(k+1,i+1)
            nums.pop()
back(0,0)