[BAEKJOON 백준] 15650 N과 M (2) (PYTHON)

 

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

15650번: N과 M (2)

문제

자연수 N과 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오.

• 1부터 N까지 자연수 중에서 중복 없이 M개를 고른 수열
• 고른 수열은 오름차순이어야 한다.입력
• 첫째 줄에 자연수 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)
• 출력
• 한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다.
• 수열은 사전 순으로 증가하는 순서로 출력해야 한다.

 

첫번째 풀이로는 dfs를 활용을 해서 문제를 해결했다. 하나의 리스트가 만들어 질때마다 출력할 수 있도록 해서 문제를 해결했다.

def dfs(start):
    if len(arr)==M:
        print(' '.join(map(str,arr)))
        return
    
    for i in range(start,N+1):
        if i not in arr:
            arr.append(i)
            dfs(i+1)
            arr.pop()


N,M=map(int,input().split())
arr = []

dfs(1)

 

itertools를 활용을 해서 combinations를 활용해 문제를 해결했다. 조합으로 중복을 제거한 배열을 만들어 주게 된다.

import itertools

N,M=map(int,input().split())
arr = list(itertools.combinations([i for i in range(1,N+1)],M))

for i in arr:
    answer=""
    for j in i:
        answer+=str(j)
        answer+=" "
    print(answer)