오뚝이개발자

[백준10974] 모든 순열 본문

코딩 테스트/백준

[백준10974] 모든 순열

땅어 2020. 3. 8. 20:52
728x90
300x250

문제


https://www.acmicpc.net/problem/10974

 

10974번: 모든 순열

N이 주어졌을 때, 1부터 N까지의 수로 이루어진 순열을 사전순으로 출력하는 프로그램을 작성하시오.

www.acmicpc.net

 

생각의 흐름


처음에 리스트를 1부터 넣어 초기화해주고 이전에 포스팅했던 다음 순열의 알고리즘을 사용하면 된다

 

 

코드


def next_permutation(arr):

    n = len(arr)-1
    if n==0:
        return -1
    i = n
    while arr[i-1] > arr[i]:
        i -= 1
        if i == 0:
            return -1

    j = i-1
    diff = 10000
    position_to_switch = 0
    for k in range(i, n+1):
        if arr[k]-arr[j] > 0:
            diff = min(diff, arr[k]-arr[j])
    for m in range(i, n+1):
        if arr[m]==arr[j]+diff:
            position_to_switch = m
            break
    temp = arr[j]
    arr[j] = arr[position_to_switch]
    arr[position_to_switch] = temp
    temp_list = arr[j+1:]
    temp_list.sort()
    arr = arr[:j+1]
    arr = arr + temp_list
    #arr[j+1:].sort()
    return arr

if __name__=='__main__':

    num = int(input())
    arr = []
    for i in range(num):
        arr.append(i+1)
    total_trial = 1
    for i in range(num):
        total_trial *= i+1
    for i in range(total_trial):
        for j in range(num):
            print(arr[j], end=' ')
        print('\n', end='')
        arr = next_permutation(arr)


 

728x90
300x250

'코딩 테스트 > 백준' 카테고리의 다른 글

[백준10971] 외판원 순회 2  (0) 2020.03.09
[백준10819] 차이를 최대로  (0) 2020.03.09
[백준10973] 이전 순열  (0) 2020.03.08
[백준10972] 다음 순열  (0) 2020.03.08
[백준9095] 1, 2, 3 더하기  (0) 2020.03.08
Comments