알고리즘 연습/백트래킹

[🥈2 / 백준 15665 / 파이썬] N과 M (11)

김세진 2022. 3. 9. 19:18
반응형

 

 

15665번: N과 M (11)

한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다. 수열은 사전 순으로 증가하는 순서로 출력해

www.acmicpc.net

 

문제

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

  • N개의 자연수 중에서 M개를 고른 수열
  • 같은 수를 여러 번 골라도 된다.

입력

첫째 줄에 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 7)

둘째 줄에 N개의 수가 주어진다. 입력으로 주어지는 수는 10,000보다 작거나 같은 자연수이다.

출력

한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다.

수열은 사전 순으로 증가하는 순서로 출력해야 한다.

 

예제 입력 1

3 1
4 4 2

예제 출력 1

2
4

예제 입력 2

4 2
9 7 9 1







예제 출력 2

1 1
1 7
1 9
7 1
7 7
7 9
9 1
9 7
9 9

예제 입력 3

4 4
1 1 2 2














예제 출력 3

1 1 1 1
1 1 1 2
1 1 2 1
1 1 2 2
1 2 1 1
1 2 1 2
1 2 2 1
1 2 2 2
2 1 1 1
2 1 1 2
2 1 2 1
2 1 2 2
2 2 1 1
2 2 1 2
2 2 2 1
2 2 2 2

 

풀이

 

주어진 배열의 중복을 제거한 뒤, 오름차순으로 정렬한다.

이후, M만큼 반복하며 수열의 첫 번째부터 순서대로 리스트에 담아 출력한다.

 

n,m = map(int,input().split())
arr = sorted(list(set(map(int,input().split()))))
ans = []

def bt(cnt):
    if cnt==m:
        print(*ans)
        return
    
    cnt+=1
    for i in arr:
        ans.append(i)
        bt(cnt)
        ans.pop()
bt(0)
반응형