본문 바로가기
알고리즘/[python] 백준 BOJ

[Python] 백준 15655 N과 M (6)

by Alan_Kim 2023. 1. 4.
728x90
반응형

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

 

15655번: N과 M (6)

N개의 자연수와 자연수 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오. N개의 자연수는 모두 다른 수이다. N개의 자연수 중에서 M개를 고른 수열

www.acmicpc.net

 

 

문제 해결

N개중 M개를 뽑는 모든 경우의 수를 오름차순으로 출력하는 문제

원소를 하나 하나 M개를 뽑아야 하므로 DFS나 BFS를 생각해 볼 수 있다.

 

 

CODE (DFS)

import sys
input = sys.stdin.readline

n, m = map(int, input().split())
A = list(map(int, input().split()))
A.sort()
num = []
def dfs(start):
    if len(num) == m:
        print(*num)
        return
    for i in range(start, len(A)):
        if not A[i] in num:
            num.append(A[i])
            dfs(i+1)
            num.pop()
dfs(0)
728x90
반응형

댓글