[프로그래머스 | 3단계] 줄서는 방법(순열, 백트랙킹)

문제 설명

n명의 사람이 일렬로 줄을 서고 있습니다. n명의 사람들에게는 각각 1번부터 n번까지 번호가 매겨져 있습니다. n명이 사람을 줄을 서는 방법은 여러가지 방법이 있습니다. 예를 들어서 3명의 사람이 있다면 다음과 같이 6개의 방법이 있습니다.

• [1, 2, 3]
• [1, 3, 2]
• [2, 1, 3]
• [2, 3, 1]
• [3, 1, 2]
• [3, 2, 1]

사람의 수 n과, 자연수 k가 주어질 때, 사람을 나열 하는 방법을 사전 순으로 나열 했을 때, k번째 방법을 return하는 solution 함수를 완성해주세요.

제한사항

• n은 20이하의 자연수 입니다.
• k는 n! 이하의 자연수 입니다.

---

입출력 예nkresult

3

5

[3,1,2]

입출력 예시 설명

입출력 예 #1
문제의 예시와 같습니다.

 

문제풀이

n은 20이하의 자연수로 DFS로 풀면 시간초과 ㅜㅜ

그렇지만, 프로그래머스에서 백트랙킹을 풀때 매번 전역변수 등의 문제가 있었는데 이제는 알것 같음!!

그냥 solution() 함수에 집어 넣자!!

값을 변경할때는 global을 표기해주기!

값을 복사할때는 .copy() 사용!!

def solution(n, k):
    global count
    
    result = []
    result_dic = {}
    check = [False] * (n+1)
    count = 0
    
    def DFS(L) :
        global count
        
        if L == n :
            count += 1
            temp = result.copy()            #result로 바로 가면 빽했을때 pop되어서 빈값 참조되어, 이를 복사뜸
            #print(result, end=" ")
            result_dic[count] = temp
            return 
        else :
            for i in range(1, n+1) :
                if check[i] is False :
                    check[i] = True
                    result.append(i)
                    DFS(L+1)
                    check[i] = False
                    result.pop()
    
    DFS(0)
    return result_dic[k]

#순열
#n개의 숫자를 n개 나열하는 경우의 수 다 구하기

 

import itertools도 시간초과!!

import itertools

def solution(n, k):
    data_list = []
    
    for i in range(1, n+1) :
        data_list.append(i)
        
    permutation_data = itertools.permutations(data_list, n)
    permutation_list = list(permutation_data)    
    return list(permutation_list[k-1])
    
     

#순열
#n개의 숫자를 n개 나열하는 경우의 수 다 구하기