[Python/파이썬 15652 백준] N과 M(4)

www.acmicpc.net/problem/15652

15652번: N과 M (4)

풀이

itertools 쓴거

엄청 쉽네용

import sys
from itertools import combinations_with_replacement

input=sys.stdin.readline

n,m=map(int,input().split())
lst=[i for i in range(1,n+1)]
lst=list(combinations_with_replacement(lst,m))

for i in lst:
    print(" ".join(map(str,i)))

안쓴대로 풀려는데

시간초과 나서

참고했습니다.

 

number는 n값을 안넘기도록 사용하여 1~n까지 출력하고,

selected는 배열의 인덱스로 사용했습니다.

 

import sys
from itertools import combinations_with_replacement
input=sys.stdin.readline

n,m=map(int,input().split())
lst=[0 for i in range(m)]

def go(number,selected,n,m):
    if selected==m:
        print(" ".join(map(str,lst)))
        return
    if number>n:
        return

    lst[selected]=number
    go(number,selected+1,n,m)
    lst[selected]=0
    go(number+1,selected,n,m)
go(1,0,n,m)