본문 바로가기

코딩테스트/Python

[Python] 백준 #15652 - N과 M (4)

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

 

15652번: N과 M (4)

한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다. 수열은 사전 순으로 증가하는 순서로 출력해

www.acmicpc.net

 

 

풀이

def dfs(x): # 앞의 숫자와 비교해야하므로 파라미터로 넘겨줌
    if len(s)==m: # s의 길이가 m과 같으면 답 출력
        print(*s)
        return

    for i in range(x,n+1): # 파라미터로 받은 숫자보다 같거나 커야함
        s.append(i) # 수를 더해주고
        dfs(i) # 더한 수를 파라미터로 가지고 dfs
        s.pop() # 원 상태로 돌리기 위해 pop

n,m = map(int,input().split())
s = [] #답 리스트
dfs(1)

 

 

  1. 현재 dfs for문에서 넣은 숫자보다 같거나 커야한다. 파라미터로 넘겨주어서 체크해준다.
  2. s의 길이가 m과 같으면 답 출력
  3. 파라미터로 받은 숫자보다 같거나 커야하기 때문에 x이상 n이하
  4. 수를 더해주고 더한 수를 파라미터로 가지고 dfs 호출
  5.  
    원 상태로 돌리기 위해 pop