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

 < 처음 풀 때 > 

 

조합과 중복이 섞여 있어서 까다로웠다. 함수를 만들어서 배열을 이용한 방법으로 해결하려고 했지만 시간 초과가 나왔고, combination_with_replacement 라는 신기한 중복 조합 함수를 알게 되었다.

 

< 다시 풀 때 >

combination_with_replacement 를 이용해서 어렵지 않게 풀 수 있었다.

 

 

import sys
import math
from itertools import product
from itertools import combinations_with_replacement
 
a,b = map(int, sys.stdin.readline().split(" "))
 
k = [ ]
 
for i in range(1,a+1):
    k.append(i)
 
q = list(combinations_with_replacement(k,b))
 
for i in q:
    print(*i)