S3 15654번: N과 M (5)

문제 보기 :  15654번: N과 M (5)

---

문제

• 정답률 : 72%

---

작성 코드

arr_num, n = map(int,input().split())
arr = list(map(int,input().split()))
arr.sort()
visited = [False]*arr_num

def combination(n,new):
    answer = []
    if len(new)==n:
        return [new]
    for i in range(len(arr)):
        if not visited[i]:
            visited[i] = True
            answer.extend(combination(n,new+[arr[i]]))
            visited[i]=False
    return answer

answer = combination(n,[])
for a in answer:
    print(*a, sep=' ')

---

풀이

S3 15649. N과 M (1)

S3 15650. N과 M (2)

S3 15652. N과 M (4)

위 문제들은 itertools의 permutations, combinations, product, combinations_with_replacement를 함수로 직접 구현한 코드들이다.

위 방법을 참고해서 풀이하면 쉽게 풀린다.