15665번: N과 M(11)

15665번: N과 M (11)

 

이전 문제에서 다른 점은 중복이 허용된다는 점이다.

여기서의 중복은 수열의 중복이 아니라, 같은 인덱스의 수가 들어갈 수 있다는 것이다.

 

---

# 15665번: N과 M (11)

n, m = map(int, input().split())
arr = sorted(list(map(int, input().split())))

result = []

def backTracking(depth):
    if depth == m:
        print(*result)
        return
    
    remember = 0
    for i in range(n):
        if arr[i] != remember:
            remember = arr[i]
            result.append(arr[i])
            backTracking(depth + 1)
            result.pop()

backTracking(0)

 

 

visited 리스트를 사용해 같은 인덱스의 수가 들어갈 수 있도록 하던 로직을 없애주면 된다.

그리고 backTracking함수의 인자는 depth를 넣어줘서 depth가 m과 같아질 경우 정답을 출력하고 해당 backTracking문을 빠져나온다.

파라미터 없이, result의 길이가 m과 같은 경우 정답을 출력하는 식으로 코드를 바꿀 수도 있다.