백준 / 이전 순열 / 10973

Question

문제링크
Silver 3

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

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

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

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

Input

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

1 2 3 4

Output

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

-1

Logic

기본 구조 :
1. 이 경우 permutations를 사용하면 시간 초과가 걸린다.
2. 우선 마지막 수를 기준으로 왼쪽으로 순차적으로 접근하여, '최초로 마지막 수보다 큰 수'가 나온 위치를 표시한다.
2-1. 이 때 표시한 위치가 바뀌지 않는다면, 다시 말해 숫자가 오름차순으로 되어 있다면, 순열의 첫 조합이므로 -1을 출력하고 종료한다.
2-2. 위 경우가 아니라면, 다시 뒤에서부터 탐색하여 끝자리 수보다 작다면 위치를 바꾼다.
3. 이 작업이 완료되면 맨 처음 위치를 표시한 수 이후의 숫자들을 맨 앞으로 보낸다.

Code

from sys import stdin

N = int(stdin.readline())
target = list(map(int,stdin.readline().split()))

if N==1 : print(-1)
else:
    idx=N
    for i in range(N-2,-1,-1):
        if target[i] > target[i+1] :
            idx = i
            break
    if idx==N : print(-1)
    else:
        idx2=N
        for i in range(N-1,-1,-1):
            if target[idx] > target[i]:
                target[idx],target[i] = target[i], target[idx]
                break
        target = target[:idx+1]+sorted(target[idx+1:],reverse=True)
        print(*target)