[🥈3 / 백준 10972 / 파이썬] 다음 순열

 

10972번: 다음 순열

 

문제

1부터 N까지의 수로 이루어진 순열이 있다. 이때, 사전순으로 다음에 오는 순열을 구하는 프로그램을 작성하시오.

사전 순으로 가장 앞서는 순열은 오름차순으로 이루어진 순열이고, 가장 마지막에 오는 순열은 내림차순으로 이루어진 순열이다.

N = 3인 경우에 사전순으로 순열을 나열하면 다음과 같다.

• 1, 2, 3
• 1, 3, 2
• 2, 1, 3
• 2, 3, 1
• 3, 1, 2
• 3, 2, 1

입력

첫째 줄에 N(1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄에 순열이 주어진다.

출력

첫째 줄에 입력으로 주어진 순열의 다음에 오는 순열을 출력한다. 만약, 사전순으로 마지막에 오는 순열인 경우에는 -1을 출력한다.

 

예제 입력 1  4 1 2 3 4	예제 출력 1  1 2 4 3
예제 입력 2  5 5 4 3 2 1	예제 출력 2  -1

 

풀이

 

C++의 경우 다음 순열을 구하는 함수가 있지만 파이썬은 그렇지 않기 때문에 직접 구해야 한다.

 

순열이 있는 배열을 arr이라고 하자.

알고리즘은 다음과 같다.

 

1. 순열을 순회하며 arr[i-1] < arr[i] 인 경우가 있나 찾아본다.
   1. 있다면 arr[i-1]의 인덱스를 swap이라고 기억해놓자.
   2. 없다면 내림차순 끝이므로 -1 출력.
2. 1-1 이라면 다시 arr[swap+1] 부터 탐색하며 arr[i-1]보다 큰 수가 있는지 탐색한다.
3. 있다면 해당 수와 자리를 교체해주고,  arr[swap+1] 부터 오름차순으로 정렬한다.

 

import sys
input = sys.stdin.readline

n = int(input())
arr = list(map(int,input().split()))
swap = -1
for i in range(n-1,0,-1):
    if arr[i-1] < arr[i]:
        swap = i-1
        break
else:
    print(-1)
    sys.exit()

for i in range(n-1,0,-1):
    if arr[swap] < arr[i]:
        arr[swap],arr[i] = arr[i],arr[swap]
        arr = arr[:swap+1]+sorted(arr[swap+1:])
        print(*arr)
        break