[백준 15652번] N과 M(4)

http://www.acmicpc.net/problem/15652

15652번: N과 M (4)

 

 

# 조건

• 자연수 N과 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하라
  • 1부터 N까지 자연수 중에서 M개를 고른 수열
  • 같은 수를 여러 번 골라도 된다.
  • 고른 수열은 비내림차순이어야 한다.
    • 길이가 K인 수열 A가 A1 <= A2 <= A3 .... <= Ak를 만족하면, 비내림차순

 

# 접근 방법

• 같은 수를 여러번 골라도 되므로 combinations가 아닌
• 중복 조합인 combinations_with_replacement 를 이용해준다.
• 또한 비내림차순 == 오름차순

from itertools import combinations_with_replacement

N, M = map(int, input().split())
arr = [i for i in range(1, N+1)]
result = list(combinations_with_replacement(arr, M))

for j in result:
    print(*j)