코딩 테스트/백준
[백준10973] 이전 순열
땅어
2020. 3. 8. 20:49
728x90
300x250
문제
https://www.acmicpc.net/problem/10973
10973번: 이전 순열
첫째 줄에 입력으로 주어진 순열의 이전에 오는 순열을 출력한다. 만약, 사전순으로 가장 처음에 오는 순열인 경우에는 -1을 출력한다.
www.acmicpc.net
생각의 흐름
- 다음 순열 문제의 대소비교 부분을 반대로 생각하면 된다.
코드
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 = max(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()
temp_list.reverse()
arr = arr[:j+1]
arr = arr + temp_list
#arr[j+1:].sort()
return arr
if __name__=='__main__':
num = int(input())
arr = list(map(int, input().split()))
result = next_permutation(arr)
if result == -1:
print(-1)
else:
for p in result:
print(p, end=' ')
728x90
300x250