본문 바로가기
프로그래밍/알고리즘 & 자료구조

백준 15655 : N과 M (6) with 파이썬

by Play_With 2024. 1. 21.
반응형

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

 

15655번: N과 M (6)

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

www.acmicpc.net

import sys
input=sys.stdin.readline
n,m = list(map(int,input().split()))
nums= [int(x) for x in input().split()]
nums.sort()
s = []

def dfs(st):
    if len(s)==m:
        print(' '.join(map(str,s)))
        return
    
    for i in range(st, len(nums)):
        if nums[i] not in s:
            s.append(nums[i])
            dfs(i+1)
            s.pop()
            
dfs(0) # 기준이 인덱스가 됨
반응형

댓글