[백준 파이썬 15650번]N과M_2★백트래킹★매개변수 고려

VER1.0

import sys

def dfs(start):
    if len(s) == m:
        print(' '.join(map(str , s)))
        return
    for i in range(start, n+1):
        if visited[i]:
            continue
        visited[i] = True
        s.append(i)
        dfs(i+1)
        s.pop()
        visited[i] = False

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

s = []
visited = [False for i in range(n+1)]
dfs(1)

기존코드에 start변수 추가 

 

start 매개변수를 추가하면서 

재귀함수 실행시 1씩 올라가게끔 실행

 

VER 2.0

import itertools
import sys

n, m = map(int, sys.stdin.readline().split())
nums = [i for i in range(1, n+1)]


array2 = itertools.combinations(nums,m) #중복된 조합 모두 포함

for i in array2:
    for j in i:
        print(j, end = ' ')
    print()

itertools의 combinations 모듈 활용하기

import itertools
import sys

n, m = map(int, sys.stdin.readline().split())
nums = [i for i in range(1, n+1)]

array = itertools.permutations(nums, m) #중복된 조합은 제외
#itertools의 permutations함수를 사용해서 풀이
#Permutations 는 배열에서 원하는 길이에 맞는 모든 조합을 구하는 함수이다.
for i in array:
    for j in i:
        print(j, end = ' ')
    print()
print('=======================')
array2 = itertools.combinations(nums,m) #중복된 조합 모두 포함

for i in array2:
    for j in i:
        print(j, end = ' ')
    print()
print('=======================')
array3 = itertools.combinations_with_replacement(nums,m)
#중복된 조합 모두 포함 , 1 1 / 2 2 등 본인의 조합도 포함하기
#(nums, m)
for i in array3:
    for j in i:
        print(j, end = ' ')
    print()